Update app.py

app.py

@@ -52,8 +52,8 @@ def get_video_i2vgen(image_in, prompt):
     result = client.predict(
         image_in,
         prompt,
-        #fn_index=1
-        api_name="/image_to_video"
+        fn_index=1
+        #api_name="/image_to_video"
     )
     print(result)
     return result
@@ -114,10 +114,10 @@ def infer(image_in, camera_shot, conditional_pose, prompt, style, chosen_model):
     
     iid_img = get_instantID(image_in, conditional_pose, prompt, style)
 
-    short_cap = get_short_caption(iid_img)
+    #short_cap = get_short_caption(iid_img)
     
     if chosen_model == "i2vgen-xl" :
-        video_res = get_video_i2vgen(iid_img, short_cap)
+        video_res = get_video_i2vgen(iid_img, prompt)
     elif chosen_model == "stable-video" :
         video_res = get_video_svd(image_in)