Update app.py

app.py

@@ -46,24 +46,28 @@ with open ("./MuseV/configs/model/T2I_all_model.py","r+") as scrip:
 from PIL import Image
 
 def add_new_image(image):
-    print(image)
-    image = Image.open(image)
-    img_settings = """- condition_images: ./MuseV/data/images/yongen.jpeg
+    image = Image.fromarray(image)
+    height = image.height
+    width = image.width
+    lr = width/height
+    ip_img = "${.condition_images}"
+    image.save("./img.png")
+    img_settings = f"""- condition_images: ./img.png
   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}
+  height: {height}
+  img_length_ratio: {lr}
+  ipadapter_image: {ip_img}
+  name: image
+  prompt: (masterpiece, best quality, highres:1),(1person, solo:1),(eye blinks:1.8),(head wave:1.3)
+  refer_image: {ip_img}
   video_path: null
-  width: 736"""
+  width: width"""
     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"])
+    subprocess.run(["python", "./MuseV/text2video.py", "--sd_model_name", "majicmixRealv6Fp16", "--unet_model_name", "musev", "-test_data_path", "./MuseV/configs/tasks/run.yaml", "--n_batch", "1", "--target_datas", "image", "--vae_model_path", "./MuseV/checkpoints/vae/sd-vae-ft-mse", "--time_size", "12", "--fps", "12"])
     #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"])
     return "./output.mp4"