update app.py

app.py

@@ -55,33 +55,7 @@ def transfer(image, transfer_style):
         return image
 
         
-def transfer_video(video_input, 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()
-        args.src = video_input
-        args.out = "video.mp4"
-        Predictor(args.weight).transform_video(args.src, args.out, args.batch_size, start=args.start, end=args.end)
-        return args.out
-        #anime_video = Predictor(args.weight).transform_video(video, args.batch_size, args.start, args.end)
-        #return anime_video
-    elif transfer_style == "Shinkai":
-        args = parse_args_video()
-        args.weight = 'GeneratorV2_train_photo_Shinkai_init.pt'
-        args.src = video_input
-        args.out = "video.mp4"
-        Predictor(args.weight).transform_video(args.src, args.out, args.batch_size, start=args.start, end=args.end)
-        return args.out
-    elif transfer_style == "Kon Satoshi":
-        args = parse_args_video()
-        args.weight = 'GeneratorV2_train_photo_Paprika_init.pt'
-        args.src = video_input
-        args.out = "video.mp4"
-        Predictor(args.weight).transform_video(args.src, args.out, args.batch_size, start=args.start, end=args.end)
-        return args.out
-    else:
-        return 0
+
 
 def clear_output(input_widget):
     input_widget = np.array([])
@@ -107,9 +81,9 @@ with gr.Blocks() as demo:
                 clear_video_button = gr.Button("Clear")
 
     image_button.click(transfer, inputs=[image_input,image_dropdown], outputs=image_output)
-    video_button.click(transfer_video, inputs=[video_input,video_dropdown],outputs=video_output)
+ 
     clear_image_button.click(clear_output, inputs=image_input,outputs=image_output)
-    clear_video_button.click(clear_output, inputs=video_input,outputs=video_output)
+    
 demo.launch()
 
 # 启动接口