Update app.py

app.py

@@ -10,27 +10,31 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.to("cuda")
 
 @spaces.GPU
-def generate_video(image, prompt, num_frames=25, resolution=(576, 1024)):
+def generate_video(image, prompt, num_frames=25, height=576, width=1024):
     # Generate the video
-    video_frames = pipe(prompt, image=image, num_frames=num_frames, height=resolution[0], width=resolution[1]).frames
+    video_frames = pipe(prompt, image=image, num_frames=num_frames, height=height, width=width).frames
     return video_frames
 
 # Create the Gradio interface
-iface = gr.Interface(
-    fn=generate_video,
-    inputs=[
-        gr.inputs.Image(type="pil", label="Upload Image"),
-        gr.inputs.Textbox(lines=2, placeholder="Enter prompt...", label="Prompt"),
-        gr.inputs.Slider(1, 50, step=1, default=25, label="Number of Frames"),
-        gr.inputs.Number(label="Resolution Height", default=576),
-        gr.inputs.Number(label="Resolution Width", default=1024)
-    ],
-    outputs=gr.outputs.Video(label="Generated Video"),
-    title="Image to Video with Stable Diffusion XT",
-    description="Upload an image and enter a prompt to generate a video."
-)
+with gr.Blocks() as demo:
+    gr.Markdown("## Image to Video with Stable Diffusion XT")
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="pil", label="Upload Image")
+            prompt_input = gr.Textbox(lines=2, placeholder="Enter prompt...", label="Prompt")
+            num_frames_input = gr.Slider(1, 50, step=1, default=25, label="Number of Frames")
+            height_input = gr.Number(label="Resolution Height", default=576)
+            width_input = gr.Number(label="Resolution Width", default=1024)
+            run_button = gr.Button("Generate Video")
+        with gr.Column():
+            video_output = gr.Video(label="Generated Video")
+
+    run_button.click(
+        generate_video,
+        inputs=[image_input, prompt_input, num_frames_input, height_input, width_input],
+        outputs=video_output
+    )
 
 # Launch the interface
 if __name__ == "__main__":
-    iface.launch()
-    
+    demo.launch()