YANGYYYY commited on
Commit
b2929ff
1 Parent(s): a62524f

Update app.py

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