Update app.py

app.py

@@ -1,13 +1,22 @@
-pip install accelerate
+import gradio as gr
 from diffusers import DiffusionPipeline
 
-pipeline=DiffusionPipeline.from_pretrained("THUDM/CogVideoX-2b")
+# Load the model
+pipeline = DiffusionPipeline.from_pretrained("ali-vilab/text-to-video-ms-1.7b")
 
-def generate_output():
+def generate_video(text):
+    video = pipeline(text)
+    # Assuming the output is a video file or a binary stream. Adjust accordingly.
+    return video
 
-    output = pipeline("your input prompt here")
-    return output
+# Create the Gradio interface
+interface = gr.Interface(
+    fn=generate_video,
+    inputs=gr.Textbox(lines=2, placeholder="Type your text here..."),
+    outputs=gr.Video(),
+    title="Text-to-Video Generator",
+    description="Enter some text and generate a video!"
+)
 
 if __name__ == "__main__":
-    result = generate_output()
-    print(result)
+    interface.launch()