Update app.py

app.py

@@ -43,6 +43,20 @@ with open ("./MuseV/configs/model/T2I_all_model.py","r+") as scrip:
     scrip.truncate()
     scrip.seek(0)
 
+def add_new_image(image):
+    img_settings = """- condition_images: ./MuseV/data/images/yongen.jpeg
+  eye_blinks_factor: 1.8
+  height: 1308
+  img_length_ratio: 0.957
+  ipadapter_image: ${.condition_images}
+  name: yongen
+  prompt: (masterpiece, best quality, highres:1),(1boy, solo:1),(eye blinks:1.8),(head wave:1.3)
+  refer_image: ${.condition_images}
+  video_path: null
+  width: 736"""
+    with open ("./MuseV/configs/tasks/run.yaml","r+") as configs:
+        configs.write(img_settings)
+        
 @spaces.GPU
 def run(duration=180):
     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", "yongen", "--vae_model_path", "./MuseV/checkpoints/vae/sd-vae-ft-mse", "--time_size", "12", "--fps", "12"])
@@ -50,7 +64,9 @@ def run(duration=180):
     return "./output.mp4"
 
 with gr.Blocks() as demo:
-  button = gr.Button()
+  image = gr.Image()
+  image.change(fn=add_new_image,inputs=image)
+  button = gr.Button(label="Generate Video")
   video = gr.Video()
   button.click(fn=run,outputs=video)