feat(handler): ajoute la prise en charge du modèle fine-tuné dans la prédiction

app/handler.py

@@ -1,19 +1,23 @@
 from models.zero_shot import ZeroShotModel
 from models.few_shot import FewShotModel
+from models.fine_tuned import FineTunedModel
 
 zero_shot_model = ZeroShotModel()
 few_shot_model = FewShotModel()
+fine_tuned_model = FineTunedModel()
 
 def predict(text: str, model_type: str = "zero-shot") -> str:
     if model_type == "few-shot":
         results = few_shot_model.predict(text)
-        output = "### Résultat de la classification (Few-Shot) :\n\n"
-        for label, score in results:
-            output += f"- **{label}** : {score*100:.1f}%\n"
-        return output
+        title = "Few-Shot"
+    elif model_type == "fine-tuned":
+        results = fine_tuned_model.predict(text)
+        title = "Fine-Tuned"
     else:
         results = zero_shot_model.predict(text)
-        output = "### Résultat de la classification (Zero-Shot) :\n\n"
-        for label, score in results:
-            output += f"- **{label}** : {score*100:.1f}%\n"
-        return output
+        title = "Zero-Shot"
+
+    output = f"### Résultat de la classification ({title}) :\n\n"
+    for label, score in results:
+        output += f"- **{label}** : {score*100:.1f}%\n"
+    return output