Create app.py

app.py

@@ -0,0 +1,59 @@
+import torch
+import gradio as gr
+from diffusers import CogVideoXPipeline
+from diffusers.utils import export_to_video
+
+# 1. Initialize the CogVideoX1.5‑5B‑SAT pipeline on ZeroGPU
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+pipe = CogVideoXPipeline.from_pretrained(
+    "THUDM/CogVideoX1.5-5B-SAT",      # High‑quality 10s model :contentReference[oaicite:5]{index=5}
+    torch_dtype=torch.float16
+).to(device)
+pipe.enable_model_cpu_offload()        # Memory optimization :contentReference[oaicite:6]{index=6}
+pipe.enable_vae_slicing()              # Further memory savings :contentReference[oaicite:7]{index=7}
+
+# 2. Define the video generation function
+def generate_video(prompt: str):
+    """
+    Generates a 10‑second video (160 frames at 16 fps) for the given prompt.
+    Returns the path to the saved MP4 file.
+    """
+    # 160 frames @16 fps → 10-second duration :contentReference[oaicite:8]{index=8}
+    result = pipe(
+        prompt,
+        num_frames=160,
+        fps=16,
+    )
+    frames = result.frames[0]
+    
+    # Export frames to MP4; Gradio's Video component accepts file paths
+    video_path = export_to_video(frames, "generated.mp4", fps=16)
+    return video_path
+
+# 3. Build the Gradio interface
+with gr.Blocks(title="CogVideoX1.5‑5B‑SAT Text‑to‑Video") as demo:
+    gr.Markdown(
+        """
+        # 🎥 Text‑to‑Video with CogVideoX1.5‑5B‑SAT
+        Enter a prompt and generate up to **10 s** of high‑quality video.
+        Powered by THUDM/CogVideoX1.5‑5B‑SAT on Hugging Face ZeroGPU.
+        """
+    )
+    with gr.Row():
+        prompt_input = gr.Textbox(
+            label="Video Prompt",
+            placeholder="e.g., A majestic eagle soaring over mountain peaks",
+            lines=2,
+        )
+        generate_btn = gr.Button("Generate Video")
+    video_output = gr.Video(label="Your Generated Video")
+    
+    generate_btn.click(
+        fn=generate_video,
+        inputs=prompt_input,
+        outputs=video_output,
+    )
+
+# 4. Launch (Spaces will handle server settings automatically)
+if __name__ == "__main__":
+    demo.launch()