Update app2.py

app2.py

@@ -45,7 +45,7 @@ with open ("./MuseV/configs/model/T2I_all_model.py","r+") as scrip:
 
 from PIL import Image
 
-def add_new_image(image):
+def add_new_image(prompt,image):
     image = Image.fromarray(image)
     height = image.height
     width = image.width
@@ -58,7 +58,7 @@ def add_new_image(image):
   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)
+  prompt: {prompt}
   refer_image: {ip_img}
   video_path: null
   width: {width}"""
@@ -89,12 +89,11 @@ def add_new_video(video):
         configs.seek(0)
         
 
-def run():
+def run(frames,fps):
     #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", "--motion_speed", "5.0", "--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", "image",  "--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", "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", "image",  "--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", frames, "--fps", fps])
     return "./output.mp4"
 
-
 def run_video():
     subprocess.run(["python", "./MuseV/video2video.py", "--sd_model_name", "fantasticmix_v10",  "--unet_model_name", "musev", "-test_data_path", "./MuseV/configs/tasks/example.yaml", "--output_dir", "./output",  "--n_batch", "1", "--controlnet_name", "dwpose_body_hand",  "--which2video", "video_middle",  "--target_datas",  "dance1", "--fps", "12", "--time_size", "12"])
     return "./output.mp4"
@@ -104,10 +103,13 @@ with gr.Blocks() as demo:
   subtitle1 = gr.Markdown("""Image2Vid""")
   image = gr.Image()
   button1 = gr.Button(value="Save Image")
-  button1.click(fn=add_new_image,inputs=[image])
+  frames = gr.Number(value=12)
+  fps = gr.Number(value=12)
+  prompt = gr.Text(value="(masterpiece, best quality, highres:1),(1person, solo:1),(eye blinks:1.8),(head wave:1.3)")
+  button1.click(fn=add_new_image,inputs=[prompt,image])
   button2 = gr.Button(value="Generate Img2Vid")
   video = gr.Video()
-  button2.click(fn=run,outputs=video)
+  button2.click(fn=run,inputs=[frames,fps],outputs=video)
   subtitle2 = gr.Markdown("""Vid2Vid""")
   video_in = gr.Video()
   button3 = gr.Button(value="Save Video")