Optimize: Force CPU mode and add low memory usage for HF Spaces

app.py

@@ -11,7 +11,7 @@ import os
 # Configuration
 MODEL_PATH = "models"
 MAX_LENGTH = 256
-DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+DEVICE = torch.device("cpu")  # Force CPU pour Hugging Face Spaces gratuit
 
 print("="*60)
 print("🚀 Initialisation du Détecteur de Phishing")
@@ -29,7 +29,12 @@ print(f"📥 Chargement du tokenizer depuis {MODEL_PATH}...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
 
 print(f"📥 Chargement du modèle depuis {MODEL_PATH}...")
-model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
+# Charger en mode CPU avec optimisations mémoire
+model = AutoModelForSequenceClassification.from_pretrained(
+    MODEL_PATH,
+    torch_dtype=torch.float32,  # Utiliser float32 pour compatibilité CPU
+    low_cpu_mem_usage=True  # Optimisation mémoire
+)
 model.to(DEVICE)
 model.eval()
 
@@ -234,8 +239,7 @@ if __name__ == "__main__":
     print("="*60 + "\n")
 
     demo.launch(
-        server_name="127.0.0.1",  # Accessible localement uniquement
-        server_port=7860,
-        share=False,  # Mettre True pour obtenir un lien public temporaire
-        show_error=True
+        show_error=True,
+        server_name="0.0.0.0",  # Nécessaire pour Hugging Face Spaces
+        server_port=7860
     )