Update app.py

app.py

@@ -97,9 +97,11 @@ 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():
+@spaces.GPU
+def run_video(duration=240):
     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"])
-
+    return "./output.mp4"
+    
 with gr.Blocks() as demo:
   title = gr.Markdown("""# MuseV Image2Vid & Vid2Vid """)
   subtitle1 = gr.Markdown("""Image2Vid""")