Update app.py
Browse files
app.py
CHANGED
|
@@ -62,12 +62,14 @@ def add_new_image(image):
|
|
| 62 |
refer_image: {ip_img}
|
| 63 |
video_path: null
|
| 64 |
width: width"""
|
| 65 |
-
with open ("/home/user/app/MuseV/configs/tasks/
|
| 66 |
configs.write(img_settings)
|
|
|
|
|
|
|
| 67 |
|
| 68 |
@spaces.GPU
|
| 69 |
def run(duration=180):
|
| 70 |
-
subprocess.run(["python", "./MuseV/text2video.py", "--sd_model_name", "majicmixRealv6Fp16", "--unet_model_name", "musev", "-test_data_path", "./MuseV/configs/tasks/
|
| 71 |
#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"])
|
| 72 |
return "./output.mp4"
|
| 73 |
|
|
|
|
| 62 |
refer_image: {ip_img}
|
| 63 |
video_path: null
|
| 64 |
width: width"""
|
| 65 |
+
with open ("/home/user/app/MuseV/configs/tasks/example.yaml","r+") as configs:
|
| 66 |
configs.write(img_settings)
|
| 67 |
+
configs.truncate()
|
| 68 |
+
configs.seek(0)
|
| 69 |
|
| 70 |
@spaces.GPU
|
| 71 |
def run(duration=180):
|
| 72 |
+
subprocess.run(["python", "./MuseV/text2video.py", "--sd_model_name", "majicmixRealv6Fp16", "--unet_model_name", "musev", "-test_data_path", "./MuseV/configs/tasks/example.yaml", "--n_batch", "1", "--target_datas", "image", "--vae_model_path", "./MuseV/checkpoints/vae/sd-vae-ft-mse", "--time_size", "12", "--fps", "12"])
|
| 73 |
#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"])
|
| 74 |
return "./output.mp4"
|
| 75 |
|