feat(models): ajoute un modèle de classification zero-shot basé sur bart-large-mnli

app/handler.py

@@ -1,13 +1,6 @@
-from models.base import BaseModel
+from models.zero_shot import ZeroShotModel
 
-# Exemple de modèle minimal
-class DummyModel(BaseModel):
-    def predict(self, text):
-        if "stupide" in text.lower():
-            return "toxique"
-        return "non-toxique"
-
-dummy = DummyModel()
+model = ZeroShotModel()
 
 def predict(text: str) -> str:
-    return dummy.predict(text)
+    return model.predict(text)

models/zero_shot.py

@@ -0,0 +1,11 @@
+from transformers import pipeline
+from models.base import BaseModel
+
+class ZeroShotModel(BaseModel):
+    def __init__(self):
+        self.classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+
+    def predict(self, text: str) -> str:
+        labels = ["toxique", "non-toxique"]
+        result = self.classifier(text, candidate_labels=labels)
+        return result["labels"][0]