Update app.py

app.py

@@ -50,7 +50,9 @@ def generate_video(prompt, seed, image=None):
         "cfg_for": False,
     }
     assert image is not None, "please input image"
-    kwargs["image"] = load_image(image=image).resize((320,320), Image.LANCZOS)
+    img = load_image(image=image)
+    img.resize((320,320), Image.LANCZOS)
+    kwargs["image"] = img
     #global predictor
     output = predictor.inference(kwargs)
     save_dir = f"./"
@@ -65,6 +67,21 @@ with gr.Blocks() as demo:
                 image = gr.Image(label="Upload Image", type="filepath")
                 prompt = gr.Textbox(label="Input Prompt")
                 seed = gr.Number(label="Random Seed", value=-1)
+                width = gr.Slider(
+                    label="Size",
+                    minimum=256,
+                    maximum=1024,
+                    step=16,
+                    value=448,
+                )
+                frames = gr.Slider(
+                    label="Number of Frames",
+                    minimum=16,
+                    maximum=256,
+                    step=8,
+                    value=64,
+                )
+                
             submit_button = gr.Button("Generate Video")
             output_video = gr.Video(label="Generated Video")
             submit_button.click(