Update app.py

app.py

@@ -1,50 +1,54 @@
 import gradio as gr
 import torch
-from diffusers import WanVideoPipeline
+from diffusers import SanaVideoPipeline
 from diffusers.utils import export_to_video
 import tempfile
 
-# إعداد النموذج - نسخة 1.3B هي الأخف والأقوى حالياً
-model_id = "Wan-AI/Wan2.1-T2V-1.3B"
+# إعداد النموذج: نسخة 0.6B هي الأخف في العالم حالياً
+model_id = "NVLabs/Sana-1.1-0.6B" # أو النسخة المتاحة في بيئتك
 
-print("جاري تحميل النموذج بتقنيات توفير الذاكرة...")
-# تحميل النموذج مع تحديد نوع البيانات float32 للـ CPU
-pipe = WanVideoPipeline.from_pretrained(
-    model_id,
+print("جاري تحميل نموذج Sana الذكي...")
+# تحميل النموذج بوضعية توفير الطاقة القصوى
+pipe = SanaVideoPipeline.from_pretrained(
+    model_id, 
     torch_dtype=torch.float32,
-    # هذه الخاصية تمنع تحميل المكونات إلا عند الحاجة إليها
-    low_cpu_mem_usage=True 
+    use_safetensors=True
 )
 
-# تحسين استهلاك الرام للمساحات المجانية
+# تفعيل خاصية نقل الأجزاء غير المستخدمة من الذاكرة
 pipe.enable_model_cpu_offload()
 
 def generate_video(prompt):
     if not prompt:
         return None
     
-    # السر في السرعة والقوة: 5 خطوات فقط تعطي نتيجة مذهلة في Wan2.1
-    video_frames = pipe(
+    # Sana يتميز بقدرته على إنتاج جودة عالية بـ 4-8 خطوات فقط!
+    output = pipe(
         prompt=prompt,
-        num_frames=9,           # فيديو قصير جداً للسرعة
-        num_inference_steps=5,  # عدد خطوات قليل جداً لتناسب الـ CPU
-        height=320,
+        height=320, 
         width=320,
-        guidance_scale=6.0,
+        num_frames=8,
+        num_inference_steps=4, # السرعة هنا!
+        guidance_scale=5.0,
     ).frames[0]
     
+    # حفظ النتيجة
     temp_file = tempfile.NamedTemporaryFile(suffix='.mp4', delete=False)
-    export_to_video(video_frames, temp_file.name, fps=8)
+    export_to_video(output, temp_file.name, fps=8)
     return temp_file.name
 
-# واجهة Gradio
-with gr.Blocks() as demo:
-    gr.Markdown("### ⚡ Wan2.1 Tiny: Fastest Strong Model")
+# واجهة المستخدم
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🚀 Sana Video: Ultra-Light CPU Gen")
+    gr.Markdown("هذا النموذج يعمل بتقنية Linear Attention لضمان السرعة على المساحات المجانية.")
+    
     with gr.Row():
-        input_text = gr.Textbox(label="Prompt", placeholder="A cute robot waving...")
-        btn = gr.Button("Generate (Takes 2-5 mins on CPU)")
-    output_video = gr.Video()
+        with gr.Column():
+            prompt = gr.Textbox(label="وصف الفيديو", placeholder="A cinematic shot of a golden retriever puppy playing in the snow...")
+            btn = gr.Button("توليد الفيديو (دقيقتين تقريباً)", variant="primary")
+        with gr.Column():
+            result = gr.Video(label="النتيجة")
     
-    btn.click(fn=generate_video, inputs=input_text, outputs=output_video)
+    btn.click(fn=generate_video, inputs=prompt, outputs=result)
 
 demo.launch()