Update main.py

main.py

@@ -10,23 +10,25 @@ app = FastAPI()
 
 model_id = "gijl/gemma-4-E4B-it"
 
-# تحميل المعالج والنموذج
-print("جاري تحميل النموذج...")
-processor = AutoProcessor.from_pretrained(model_id)
+# --- السطر 13: استبدال الجزء القديم بهذا الجزء الجديد ---
+print("جاري تحميل المعالج...")
+processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
+
+print("جاري تحميل النموذج (قد يستغرق وقتاً بسبب الحجم)...")
 model = AutoModelForImageTextToText.from_pretrained(
     model_id,
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
-    device_map="auto"
+    device_map="auto",
+    trust_remote_code=True 
 )
+# -------------------------------------------------------
 
-# عرض واجهة React عند الدخول للموقع
 @app.get("/")
 async def read_index():
     with open("index.html", "r", encoding="utf-8") as f:
         return HTMLResponse(content=f.read())
 
-# نقطة النهاية (Endpoint) لاستقبال الصورة والنص وإرجاع النتيجة
 @app.post("/generate")
 async def generate_text(image: UploadFile = File(...), text: str = Form(...)):
     try: