Update app.py

app.py

@@ -74,21 +74,24 @@ def run(duration=240):
     #subprocess.run(["python", "./MuseV/text2video.py", "--sd_model_name", "majicmixRealv6Fp16", "--unet_model_name", "musev_referencenet", "--referencenet_model_name", "musev_referencenet", "--ip_adapter_model_name", "musev_referencenet", "-test_data_path", "./MuseV/configs/tasks/example.yaml", "--output_dir", "./MuseV",  "--n_batch", "1",  "--target_datas", "jinkesi2",  "--vision_clip_extractor_class_name", "ImageClipVisionFeatureExtractor", "--vision_clip_model_path", "./MuseV/checkpoints/IP-Adapter/models/image_encoder", "--motion_speed", "5.0", "--vae_model_path", "./MuseV/checkpoints/vae/sd-vae-ft-mse", "--time_size", "120", "--fps", "24"])
     return "./output.mp4"
 
+def run_video():
+    subprocess.run(["python", "./MuseV/video2video.py", "--sd_model_name", "fantasticmix_v10",  "--unet_model_name", "musev", "-test_data_path", "./MuseV/configs/tasks/example.yaml", "--output_dir", "./output",  "--n_batch", "1", "--controlnet_name", "dwpose_body_hand",  "--which2video" "\"video_middle\"",  "--target_datas",  "dance1",   "--fps 12", "--time_size 12"])
+
 with gr.Blocks() as demo:
   title = gr.Markdown("""# MuseV Image2Vid & Vid2Vid """)
   subtitle1 = gr.Markdown("""Image2Vid""")
   image = gr.Image()
   button1 = gr.Button(value="Save Image")
-  button1.click(fn=add_new_image,inputs=image)
+  button1.click(fn=add_new_image,inputs=[image])
   button2 = gr.Button(value="Generate Img2Vid")
   video = gr.Video()
   button2.click(fn=run,outputs=video)
   subtitle2 = gr.Markdown("""Vid2Vid""")
   video_in = gr.Video()
-  button2 = gr.Button(value="Generate Vid2Vid")
+  button3 = gr.Button(value="Save Video")
+  button3.click(fn=add_new_video,inputs=[video_in])
+  button4 = gr.Button(value="Generate Vid2Vid")
   video_out = gr.Video()
-  button2.click(fn=run,outputs=video_out)
-  
-    
+  button4.click(fn=run_video,outputs=video_out)  
 
 demo.launch()