Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def transfer_video(video, transfer_style):
|
|
59 |
#output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
|
60 |
#os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
|
61 |
args = parse_args_video()
|
62 |
-
anime_video = Predictor(args.weight).transform_video(video, args.batch_size
|
63 |
return anime_video
|
64 |
elif transfer_style == "Shinkai":
|
65 |
args = parse_args()
|
|
|
59 |
#output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
|
60 |
#os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
|
61 |
args = parse_args_video()
|
62 |
+
anime_video = Predictor(args.weight).transform_video(video, args.batch_size)
|
63 |
return anime_video
|
64 |
elif transfer_style == "Shinkai":
|
65 |
args = parse_args()
|