Update gradio_app.py

gradio_app.py

@@ -125,10 +125,11 @@ def untranspose(tensor):
     ndim = tensor.ndim
     return tensor.transpose(ndim-1, ndim-2)
 
+[i2v_input_image, i2v_input_text, i2v_input_image, i2v_input_image2,  i2v_steps, i2v_eta, i2v_motion, i2v_seed],
 @spaces.GPU(duration=200)
-def get_image(model, prompts, image1, image2, noise_shape, n_samples=1, ddim_steps=50, ddim_eta=1., \
-                        unconditional_guidance_scale=1.0, cfg_img=None, fs=None, seed=123, text_input=False, multiple_cond_cfg=False, \
-                           loop=False, interp=False, timestep_spacing='uniform', guidance_rescale=0.0, **kwargs):
+def get_image(image1, prompt, image2, dim_steps=50, ddim_eta=1., fs=None, seed=123, \
+                        unconditional_guidance_scale=1.0, cfg_img=None,  text_input=False, multiple_cond_cfg=False, \
+                        loop=False, interp=False, timestep_spacing='uniform', guidance_rescale=0.0, noise_shape, n_samples=1, **kwargs):
     
     seed_everything(seed)
     # image1 = Image.open(file_list[2*idx]).convert('RGB')
@@ -313,7 +314,9 @@ def dynamicrafter_demo(result_dir='./tmp/', res=1024):
                 #             fn = get_image,
                 #             cache_examples=False,
                 # )
-            i2v_end_btn.click(inputs=[i2v_input_image, i2v_input_text, i2v_input_image, i2v_input_image2, [72, 108], 1, i2v_steps, i2v_eta, 1.0, None, i2v_motion, i2v_seed],
+            img_size = [72, 108]
+            
+            i2v_end_btn.click(inputs=[i2v_input_image, i2v_input_text, i2v_input_image2, i2v_steps, i2v_eta, i2v_motion, i2v_seed],
                             outputs=[i2v_output_video],
                             fn = get_image
             )