Update setup.sh

setup.sh

@@ -6,11 +6,11 @@ if [ -z "$HF_TOKEN" ]; then
     echo "⚠️ متغير HF_TOKEN غير معرف. قد يتسبب هذا في فشل تحميل النموذج."
 fi
 
-# إنشاء مجلد لتخزين النموذج
-mkdir -p models
-chown -R appuser:appuser models
+# إنشاء مجلدات لتخزين النماذج والـ cache
+mkdir -p models /app/.cache/huggingface
+chown -R appuser:appuser models /app/.cache/huggingface
 
-# تحميل ملف .gguf إذا لم يكن موجودًا مسبقًا
+# تحميل ملف Mistral .gguf إذا لم يكن موجودًا مسبقًا
 python - <<PY
 from huggingface_hub import hf_hub_download
 import os
@@ -28,5 +28,24 @@ if not os.path.exists(os.path.join(local_dir, filename)):
     )
     print("✅ تم تحميل Mistral .gguf")
 else:
-    print("✅ الملف موجود مسبقًا")
+    print("✅ ملف Mistral .gguf موجود مسبقًا")
 PY
+
+# تحميل نموذج MGZON-FLAN-T5 إذا لم يكن موجودًا مسبقًا
+python - <<PY
+from huggingface_hub import snapshot_download
+import os
+repo_id   = "MGZON/mgzon-flan-t5-base"
+local_dir = "/app/.cache/huggingface/hub/models--MGZON--mgzon-flan-t5-base/snapshots"
+if not os.path.exists(local_dir):
+    snapshot_download(
+        repo_id=repo_id,
+        local_dir=local_dir,
+        local_dir_use_symlinks=False,
+        force_download=False,
+        token=os.environ.get("HF_TOKEN")
+    )
+    print("✅ تم تحميل MGZON-FLAN-T5")
+else:
+    print("✅ ملفات MGZON-FLAN-T5 موجودة مسبقًا")
+PY