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 |
|