YANGYYYY commited on
Commit
d6166cf
1 Parent(s): 9069025

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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