Update app.py

app.py

@@ -1,35 +1,40 @@
 import os
-from huggingface_hub import InferenceClient
 import gradio as gr
+from huggingface_hub import InferenceClient
 
-# ———————— CONFIG ————————
-# (You’ve already set HF_HUB_TOKEN in your Space’s Secrets UI)
-hf_token = os.environ["HF_HUB_TOKEN"]
+# ─── STEP 1: read your token from the Spaces secrets
+HF_TOKEN = os.environ.get("HF_HUB_TOKEN")
+if HF_TOKEN is None:
+    raise RuntimeError("❌  Please add your HF_HUB_TOKEN under Settings → Variables and secrets")
 
-# Initialize the client against the text-to-video model
-client = InferenceClient(
-    model="ali-vilab/text-to-video-ms-1.7b",
-    token=hf_token,
-)
+# ─── STEP 2: initialize the client (no repo_id arg any more)
+client = InferenceClient(token=HF_TOKEN)
 
-# ———————— LOGIC ————————
+# ─── STEP 3: the generation function
 def generate_video(prompt: str) -> str:
-    # call the HF text-to-video endpoint
-    res = client.text_to_video(prompt)       # ➞ {'video': b'...'}
-    video_bytes = res["video"]
-    out_path = "out.mp4"
-    with open(out_path, "wb") as f:
-        f.write(video_bytes)
-    return out_path
+    try:
+        # pass your model ID as the first arg, and the text as `inputs=…`
+        result = client.text_to_video(
+            "ali-vilab/text-to-video-ms-1.7b", 
+            inputs=prompt
+        )
+        video_bytes = result["video"]
+        out_path = "output.mp4"
+        with open(out_path, "wb") as f:
+            f.write(video_bytes)
+        return out_path
+    except Exception as e:
+        # re-raise so Gradio logs it for you
+        raise RuntimeError(f"Video generation failed: {e}")
 
-# ———————— UI ————————
-demo = gr.Interface(
-    fn=generate_video,
-    inputs=gr.Text(label="Enter your prompt here"),
-    outputs=gr.Video(label="Generated Video"),
-    title="🎬 Video Generator",
-    description="Generate a short MP4 from your text prompt via the HF Inference API"
-)
+# ─── STEP 4: wire it up in Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎬 Video Generator")
+    with gr.Row():
+        inp = gr.Textbox(placeholder="Type your prompt here", label="Enter your prompt")
+        vid = gr.Video(label="Generated Video")
+    btn = gr.Button("Generate")
+    btn.click(fn=generate_video, inputs=inp, outputs=vid)
 
-if __name__ == "__main__":
-    demo.launch()
+# ─── STEP 5: launch
+demo.launch()