fffiloni commited on
Commit
3af7001
1 Parent(s): 92fe2b4

add moondream step

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -95,10 +95,10 @@ def infer(image_in, camera_shot, conditional_pose, prompt, style, chosen_model):
95
 
96
  iid_img = get_instantID(image_in, conditional_pose, prompt, style)
97
 
98
- #short_cap = get_short_caption(iid_img)
99
 
100
  if chosen_model == "i2vgen-xl" :
101
- video_res = get_video_i2vgen(iid_img, prompt)
102
  elif chosen_model == "stable-video" :
103
  video_res = get_video_svd(image_in)
104
 
 
95
 
96
  iid_img = get_instantID(image_in, conditional_pose, prompt, style)
97
 
98
+ short_cap = get_short_caption(iid_img)
99
 
100
  if chosen_model == "i2vgen-xl" :
101
+ video_res = get_video_i2vgen(iid_img, short_cap)
102
  elif chosen_model == "stable-video" :
103
  video_res = get_video_svd(image_in)
104