Update app.py

app.py

@@ -59,23 +59,23 @@ def transfer_video(video, transfer_style):
     if transfer_style == "Hayao":
          #output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 转换为灰度图像
         #os.system("wget https://huggingface.co/YANGYYYY/cartoonize/resolve/main/GeneratorV2_train_photo_Hayao_init.pt")
-        args = parse_args_video()
-        # 加载视频文件
-        #video_binary = io.BytesIO(video)
+        # args = parse_args_video()
+        # # 加载视频文件
+        # #video_binary = io.BytesIO(video)
             
-        cap = cv2.VideoCapture(video)
+        # cap = cv2.VideoCapture(video)
         
-        # 读取视频帧并保存到一个列表中
-        video_frames = []
-        while True:
-            ret, frame = cap.read()
-            if not ret:
-                break
-            video_frames.append(frame)
+        # # 读取视频帧并保存到一个列表中
+        # video_frames = []
+        # while True:
+        #     ret, frame = cap.read()
+        #     if not ret:
+        #         break
+        #     video_frames.append(frame)
         
-        # 关闭视频文件
-        cap.release()
-        anime_video = Predictor(args.weight).transform_video(video_frames, args.batch_size)
+        # # 关闭视频文件
+        # cap.release()
+        anime_video = Predictor(args.weight).transform_video(video, args.batch_size)
         return anime_video
     elif transfer_style == "Shinkai":
         args = parse_args()