Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def transfer(image, transfer_style):
|
|
25 |
#os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
|
26 |
args = parse_args()
|
27 |
predictor = Predictor(args.weight, args.device)
|
28 |
-
anime_img = predictor.transform_image(
|
29 |
return anime_img
|
30 |
elif transfer_style == "Shinkai":
|
31 |
return image
|
|
|
25 |
#os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
|
26 |
args = parse_args()
|
27 |
predictor = Predictor(args.weight, args.device)
|
28 |
+
anime_img = predictor.transform_image(image)
|
29 |
return anime_img
|
30 |
elif transfer_style == "Shinkai":
|
31 |
return image
|