Update handler.py

handler.py

@@ -17,12 +17,15 @@ class EndpointHandler:
         Initialise le handler avec le modèle haywoodsloan/ai-image-detector-deploy
         et configure Grad-CAM pour les cartes de saillance.
         """
-        print(f"Initialisation du handler avec le modèle : {model_dir}")
+        # Forcer l'utilisation du modèle Hugging Face Hub même lors du déploiement
+        model_name = "haywoodsloan/ai-image-detector-deploy"
+        print(f"Initialisation du handler avec le modèle : {model_name}")
+        print(f"Répertoire de déploiement : {model_dir}")
         
-        # Charger le modèle et le processeur
-        self.model = AutoModelForImageClassification.from_pretrained(model_dir)
+        # Charger le modèle et le processeur depuis Hugging Face Hub
+        self.model = AutoModelForImageClassification.from_pretrained(model_name)
         self.model.eval()
-        self.processor = AutoImageProcessor.from_pretrained(model_dir)
+        self.processor = AutoImageProcessor.from_pretrained(model_name)
         
         # Configuration de Grad-CAM - Adaptation pour différentes architectures
         # Le modèle haywoodsloan/ai-image-detector-deploy est généralement basé sur ViT ou CNN