Update app.py

app.py

@@ -1,15 +1,11 @@
 from diffusers import DiffusionPipeline
-import torch
 
-device = "cuda" if torch.cuda.is _available() else "cpu"
+pipeline=DiffusionPipeline.from_pretrained("THUDM/CogVideoX-2b")
 
-pipeline=DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
-pipeline = pipeline.to(device)
+def generate_output()
 
-prompt = "A beautiful sunset over the ocean"
-video = pipeline(prompt,num_interface_steps=50).videos
+output = pipeline("your input prompt here")return output
 
-video path = "generated_video.mp4"with open(video_path, "wb") as f:
-    f.write(video[0])
-
-print(f"video generated and saved as {video_path}")
+if __name__ == "__main__":
+    result = generate_output()
+    print(result)