Update app.py

app.py

@@ -63,7 +63,7 @@ def generate_video(prompt, seed, image, size, steps, frames, progress=gr.Progres
     }
     assert image is not None, "please input image"
     img = load_image(image=image)
-    img.resize((320,320), Image.LANCZOS)
+    img.resize((size,size), Image.LANCZOS)
     kwargs["image"] = img
     output = predictor.inference(kwargs)
     save_dir = f"./"