Update app.py

app.py

@@ -55,10 +55,9 @@ logger = logging.getLogger(__name__)
 CACHE_DIR = "/tmp/huggingface_cache"
 os.makedirs(CACHE_DIR, exist_ok=True)
 
-# تكوين متغيرات البيئة لـ Hugging Face
+# تكوين متغيرات البيئة لـ Hugging Face (removed TRANSFORMERS_CACHE to avoid deprecation warning)
 os.environ.update({
     "HF_HOME": CACHE_DIR,
-    "TRANSFORMERS_CACHE": CACHE_DIR,
     "HF_DATASETS_CACHE": CACHE_DIR,
     "HUGGINGFACE_HUB_CACHE": CACHE_DIR,
     "TORCH_HOME": CACHE_DIR,
@@ -166,7 +165,7 @@ class ModelManager:
                 self.base_model_id,
                 num_labels=41,
                 cache_dir=CACHE_DIR,
-                torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+                dtype=torch.float16 if torch.cuda.is_available() else torch.float32,  # Updated from torch_dtype
                 low_cpu_mem_usage=True,
                 trust_remote_code=False
             )
@@ -180,7 +179,7 @@ class ModelManager:
                     self.fallback_model_id,
                     num_labels=41,
                     cache_dir=CACHE_DIR,
-                    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+                    dtype=torch.float16 if torch.cuda.is_available() else torch.float32,  # Updated from torch_dtype
                     low_cpu_mem_usage=True,
                     trust_remote_code=False
                 )
@@ -235,7 +234,7 @@ class ModelManager:
         logger.info(f"✅ {model_name} loaded successfully (fallback: {self.using_fallback})")
         return model
         
-    def load_models(self, max_models=2):
+    def load_models(self, max_models=3):  # Increased default to 3 to load local + 2 URLs
         """تحميل الموديلات بحد أقصى للذاكرة"""
         if self.models_loaded:
             logger.info("✨ Models already loaded")
@@ -394,7 +393,7 @@ def split_into_paragraphs(text: str) -> List[str]:
 app = FastAPI(
     title="ModernBERT AI Text Detector",
     description="كشف النصوص المكتوبة بواسطة الذكاء الاصطناعي",
-    version="2.2.0"  # Updated version with UID fix
+    version="2.3.0"  # Updated version with 3 models and deprecation fixes
 )
 
 # إضافة CORS للسماح بالاستخدام من المتصفح
@@ -441,11 +440,11 @@ async def startup_event():
     logger.info("=" * 50)
     
     # محاولة تحميل الموديلات
-    max_models = int(os.environ.get("MAX_MODELS", "2"))
+    max_models = int(os.environ.get("MAX_MODELS", "3"))  # Updated default to 3
     success = model_manager.load_models(max_models=max_models)
     
     if success:
-        logger.info("✅ Application ready! (Fallback mode: %s)", model_manager.using_fallback)
+        logger.info(f"✅ Application ready! (Fallback mode: {model_manager.using_fallback})")
     else:
         logger.error("⚠️ Failed to load models - API will return errors")
         logger.info("💡 Tip: Ensure 'transformers>=4.45.0' and 'huggingface_hub' are installed. Run: pip install --upgrade transformers huggingface_hub")