Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ def transfer(image, transfer_style):
|
|
17 |
if transfer_style == "Hayao":
|
18 |
#output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
|
19 |
os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
|
|
|
20 |
predictor = Predictor(args.weight, args.device)
|
21 |
anime_img = predictor.transform_image(args.src, save_path)
|
22 |
return anime_img
|
|
|
17 |
if transfer_style == "Hayao":
|
18 |
#output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
|
19 |
os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
|
20 |
+
args = parse_args()
|
21 |
predictor = Predictor(args.weight, args.device)
|
22 |
anime_img = predictor.transform_image(args.src, save_path)
|
23 |
return anime_img
|