Update app.py

app.py

@@ -13,30 +13,37 @@ pipe.enable_model_cpu_offload()
 pipe.vae.enable_slicing()
 
 # 2. GPU-decorated generation function
-@spaces.GPU(duration=180)
-def generate_video(prompt: str, steps: int, frames: int, fps: int) -> str:
+@spaces.GPU(duration=600)
+def generate_video(prompt: str, steps: int, frames: int, fps: int, resolution: str) -> str:
     """
-    Generates video from text with adjustable steps, frames, and fps.
+    Generates video from text with adjustable steps, frames, fps, and resolution.
     Returns path to the saved MP4 file.
     """
+    # Parse resolution string (e.g., "720p" -> height=720)
+    height = int(resolution.rstrip('p'))
+    # Width is computed to maintain the model's aspect ratio (assumed 1360x768 -> 16:9)
+    width = int(height * (16/9))
+
     # Run pipeline; offload handles device placement
     output = pipe(
         prompt=prompt,
         num_inference_steps=steps,
-        num_frames=frames
+        num_frames=frames,
+        height=height,
+        width=width
     )
     frame_list = output.frames[0]
 
     # Export to MP4 for browser playback
     return export_to_video(frame_list, "generated.mp4", fps=fps)
 
-# 3. Build the Gradio interface with sliders
+# 3. Build the Gradio interface with sliders and resolution dropdown
 with gr.Blocks(title="CogVideoX Interactive Text-to-Video") as demo:
     gr.Markdown(
         """
         # 🎞️ Interactive Text‑to‑Video Demo
-        Adjust the sliders below to control the number of diffusion steps,
-        total frames (length), and frames per second (fps) for your video.
+        Adjust the sliders and select resolution to control the diffusion steps,
+        total frames (length), fps, and video resolution.
         """
     )
     with gr.Column():
@@ -57,12 +64,17 @@ with gr.Blocks(title="CogVideoX Interactive Text-to-Video") as demo:
             minimum=1, maximum=60, step=1, value=16,
             label="Frames per Second (fps)"
         )
+        resolution_dropdown = gr.Dropdown(
+            choices=["360p", "480p", "720p", "1080p"],
+            value="480p",
+            label="Resolution"
+        )
         gen_button = gr.Button("Generate Video")
     video_output = gr.Video(label="Generated Video", format="mp4")
 
     gen_button.click(
         fn=generate_video,
-        inputs=[prompt_input, steps_slider, frames_slider, fps_slider],
+        inputs=[prompt_input, steps_slider, frames_slider, fps_slider, resolution_dropdown],
         outputs=video_output
     )