Update app.py

app.py

@@ -1,16 +1,12 @@
+from diffusers import DiffusionPipeline
 import gradio as gr
 import torch
-from diffusers import AnimateDiffPipeline, DDIMScheduler
 
-# Load AnimateDiff model (based on SDXL)
-model_id = "cerspense/animation-diffusion-sdxl"
-pipe = AnimateDiffPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
-pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
+pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dtype=torch.float16)
+pipe.to("cuda")
 
-def generate_video(prompt):
-    video = pipe(prompt, num_frames=16, guidance_scale=7.5).frames
-    return video[0]
+def generate(prompt):
+    video_frames = pipe(prompt, num_inference_steps=25).frames
+    return video_frames[0]  # return first frame or full video logic
 
-gr.Interface(fn=generate_video,
-             inputs=gr.Textbox(label="Enter your prompt"),
-             outputs=gr.Video(label="Generated Video")).launch()
+gr.Interface(fn=generate, inputs="text", outputs="image").launch()