Update app.py

app.py

@@ -1,48 +1,34 @@
-# app.py
 import os
-import gradio as gr
 from huggingface_hub import InferenceClient
+import gradio as gr
 
-# ── CONFIG ─────────────────────────────────────────────────────────────────────
-
-# (1) Your HF token must be set in Secrets as HF_HUB_TOKEN
-HF_TOKEN = os.getenv("HF_HUB_TOKEN", None)
-if not HF_TOKEN:
-    raise RuntimeError("Please set a HF_HUB_TOKEN in your Space secrets!")
-
-# (2) Model to call
-MODEL_ID = "damo-vilab/text-to-video-ms-1.7b"
-
-# Initialize the HF Inference client
-client = InferenceClient(token=HF_TOKEN)
+# ─── STEP 1 ───
+# Paste your HF token here (bypasses the need for the Spaces secrets UI)
+os.environ["HF_HUB_TOKEN"] = "YOUR_HF_TOKEN_HERE"
 
-# ── GENERATION FUNCTION ────────────────────────────────────────────────────────
+# ─── STEP 2 ───
+# Initialize the InferenceClient
+client = InferenceClient(
+    repo_id="ali-vilab/text-to-video-ms-1.7b",
+    token=os.environ["HF_HUB_TOKEN"],
+)
 
+# ─── STEP 3 ───
+# Define the function that takes a prompt, calls the API, and writes out an MP4
 def generate_video(prompt: str) -> str:
-    """
-    Calls HF's text-to-video API, writes out an MP4, and returns its path.
-    """
-    out_path = "/tmp/out.mp4"
-
-    # 1) Fire off the inference request
-    # The `text_to_video` method returns raw bytes for the first generated video.
-    result = client.text_to_video(model=MODEL_ID, inputs=prompt)
-
-    # 2) Save to disk
-    video_bytes = result["generated_video"]  # bytes
+    result = client.text_to_video(prompt)       # {'video': b'...'}
+    video_bytes = result["video"]
+    out_path = "out.mp4"
     with open(out_path, "wb") as f:
         f.write(video_bytes)
-
     return out_path
 
-# ── GRADIO UI ─────────────────────────────────────────────────────────────────
-
-with gr.Blocks() as demo:
-    gr.Markdown("## 🎬 Text-to-Video Generator")
-    prompt = gr.Textbox(label="Enter your prompt here", placeholder="a chicken crosses the road")
-    video_out = gr.Video(label="Generated Video")
-    btn = gr.Button("Generate Video")
-
-    btn.click(fn=generate_video, inputs=prompt, outputs=video_out)
-
-demo.launch(share=True)
+# ─── STEP 4 ───
+# Wire it up in Gradio
+demo = gr.Interface(
+    fn=generate_video,
+    inputs=gr.Textbox(label="Enter your prompt here"),
+    outputs=gr.Video(label="Generated Video"),
+    title="Video Generator",
+)
+demo.launch(server_name="0.0.0.0", server_port=7860)