Update app.py

app.py

@@ -28,9 +28,17 @@ def transfer(image, transfer_style):
         anime_img = predictor.transform_image(image)
         return anime_img
     elif transfer_style == "Shinkai":
-        return image
+        args = parse_args()
+        args.weight = 'GeneratorV2_train_photo_Shinkai_init.pt'
+        predictor = Predictor(args.weight, args.device)
+        anime_img = predictor.transform_image(image)
+        return anime_img
     elif transfer_style == "Kon Satoshi":
-        return image
+        args = parse_args()
+        args.weight = 'GeneratorV2_train_photo_Paprika_init.pt'
+        predictor = Predictor(args.weight, args.device)
+        anime_img = predictor.transform_image(image)
+        return anime_img
     else:
         return image