Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, UploadFile, File, Query
-from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 from huggingface_hub import hf_hub_download
 from fastapi.responses import StreamingResponse
 from PIL import Image
@@ -24,19 +24,22 @@ pipe = StableDiffusionPipeline.from_single_file(
     requires_safety_checker=False
 ).to("cpu")
 
+# ---------- استخدام Scheduler أسرع (DPM-Solver) ----------
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+
 # ---------- تحميل الـ LoRA ودمجه ----------
 print("Downloading LoRA...")
 lora_path = hf_hub_download(repo_id=MODEL_REPO, filename=LORA_FILE)
 
 print("Loading LoRA...")
 pipe.load_lora_weights(lora_path)
-pipe.fuse_lora(lora_scale=1.0)  # القوة الكاملة للـ LoRA
+pipe.fuse_lora(lora_scale=1.0)  # القوة الكاملة
 
-# تحسين الأداء (بدون تغيير الجدول الزمني)
+# تحسينات الذاكرة
 pipe.enable_attention_slicing()
 pipe.enable_vae_slicing()
 
-print("Model ready 🔥 (Base + LoRA fused)")
+print("Model ready 🔥 (Base + LoRA fused with DPM-Solver)")
 
 # ---------- دوال مساعدة ----------
 def to_bytes(img: Image.Image) -> io.BytesIO:
@@ -45,14 +48,14 @@ def to_bytes(img: Image.Image) -> io.BytesIO:
     buf.seek(0)
     return buf
 
-# ---------- Text‑to‑Image (GET بسيط بباراميتر تيكست) ----------
+# ---------- Text‑to‑Image (GET بسيط) ----------
 @app.get("/txt2img")
 def txt2img(
     prompt: str = Query(..., description="الوصف اللي عايز تولده"),
-    steps: int = Query(8, ge=1, le=20),
+    steps: int = Query(6, ge=1, le=20),            # افتراضي 6 (أسرع)
     guidance: float = Query(7.5, ge=1.0, le=20.0),
-    height: int = Query(512, ge=256, le=768),
-    width: int = Query(512, ge=256, le=768)
+    height: int = Query(384, ge=256, le=768),      # افتراضي 384
+    width: int = Query(384, ge=256, le=768)        # افتراضي 384
 ):
     image = pipe(
         prompt=prompt,
@@ -63,16 +66,16 @@ def txt2img(
     ).images[0]
     return StreamingResponse(to_bytes(image), media_type="image/png")
 
-# ---------- Image‑to‑Image (POST مع رفع صورة + باراميترات) ----------
+# ---------- Image‑to‑Image (POST) ----------
 @app.post("/img2img")
 async def img2img(
     file: UploadFile = File(...),
     prompt: str = Query(""),
-    steps: int = Query(8, ge=1, le=20),
+    steps: int = Query(6, ge=1, le=20),
     guidance: float = Query(7.5, ge=1.0, le=20.0),
     strength: float = Query(0.6, ge=0.0, le=1.0),
-    height: int = Query(512, ge=256, le=768),
-    width: int = Query(512, ge=256, le=768)
+    height: int = Query(384, ge=256, le=768),
+    width: int = Query(384, ge=256, le=768)
 ):
     img = Image.open(io.BytesIO(await file.read())).convert("RGB").resize((width, height))
     image = pipe(