YANGYYYY commited on
Commit
ead2435
1 Parent(s): f3bbb27

Update app.py

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