talalif commited on
Commit
f8b1e61
·
verified ·
1 Parent(s): d62198c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import subprocess
2
  import os
3
  import gradio as gr
 
4
 
5
  subprocess.run(["git", "clone", "https://github.com/fat-ai/MuseV.git"])
6
 
@@ -19,6 +20,7 @@ command = "\"import sys; sys.path.append('./MuseV/MuseV'); sys.path.append('./Mu
19
  subprocess.run(["python","-c",command])
20
  subprocess.run(["mv", "./MuseV/scripts/inference/text2video.py", "./MuseV/text2video.py"])
21
 
 
22
  def run():
23
  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", "./configs/tasks/example.yaml", "--output_dir", "./output", "--n_batch", "1", "--target_datas", "Mona_Lisa", "--vision_clip_extractor_class_name", "ImageClipVisionFeatureExtractor", "--vision_clip_model_path", "./checkpoints/IP-Adapter/models/image_encoder", "--motion_speed", "5.0", "--time_size", "120", "--fps", "24"])
24
 
 
1
  import subprocess
2
  import os
3
  import gradio as gr
4
+ import spaces
5
 
6
  subprocess.run(["git", "clone", "https://github.com/fat-ai/MuseV.git"])
7
 
 
20
  subprocess.run(["python","-c",command])
21
  subprocess.run(["mv", "./MuseV/scripts/inference/text2video.py", "./MuseV/text2video.py"])
22
 
23
+ @spaces.GPU
24
  def run():
25
  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", "./configs/tasks/example.yaml", "--output_dir", "./output", "--n_batch", "1", "--target_datas", "Mona_Lisa", "--vision_clip_extractor_class_name", "ImageClipVisionFeatureExtractor", "--vision_clip_model_path", "./checkpoints/IP-Adapter/models/image_encoder", "--motion_speed", "5.0", "--time_size", "120", "--fps", "24"])
26