Update app.py
Browse files
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 |
|