Update app.py

app.py

@@ -1,32 +1,23 @@
 import gradio as gr
 import torch
-from diffusers import AnimateDiffPipeline
-from PIL import Image
-import imageio
+from diffusers import DiffusionPipeline
 
-model_id = "luodian/AnimateDiff"
-
-pipe = AnimateDiffPipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16,
-    variant="fp16",
-    use_auth_token=True
-)
+model_id = "cerspense/zeroscope_v2_576w"
+pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-def image_to_video(image, prompt):
-    image = image.convert("RGB").resize((512, 512))
-    result = pipe(prompt=prompt, image=image, num_frames=16, guidance_scale=7.5).frames
+def text_to_video(prompt):
+    output = pipe(prompt, num_inference_steps=25)
+    video_frames = output.frames[0]
     output_path = "output.mp4"
-    imageio.mimsave(output_path, result, fps=8)
+
+    import imageio
+    imageio.mimsave(output_path, video_frames, fps=8)
     return output_path
 
 gr.Interface(
-    fn=image_to_video,
-    inputs=[
-        gr.Image(type="pil", label="ارفع صورة (512x512)"),
-        gr.Textbox(label="أدخل وصف الفيديو (اختياري)")
-    ],
+    fn=text_to_video,
+    inputs=gr.Textbox(label="أدخل وصف الفيديو (بالإنجليزية)"),
     outputs=gr.Video(label="الفيديو الناتج"),
-    title="تحويل صورة إلى فيديو متحرك باستخدام AnimateDiff"
-).launch()
+    title="تحويل نص إلى فيديو باستخدام ZeroScope"
+).launch()