Update app.py

app.py

@@ -5,17 +5,21 @@ import torch
 # Agent 1: Intent Classifier
 intent_classifier = pipeline("text-classification", model="MoritzLaurer/DeBERTa-v3-base-mnli-fever-anli")
 
+
 def detect_intent(text):
     labels = {
         "weather": "The user wants to know the weather.",
         "faq": "The user is asking for help.",
         "smalltalk": "The user is making casual conversation."
     }
-    scores = {}
+    best_intent = "smalltalk"
+    best_score = 0
     for label, hypothesis in labels.items():
-        result = intent_classifier({"premise": text, "hypothesis": hypothesis})[0]
-        scores[label] = result['score'] if result['label'] == 'ENTAILMENT' else 0
-    return max(scores, key=scores.get)
+        result = intent_classifier(text=text, text_pair=hypothesis)[0]
+        if result['label'] == 'ENTAILMENT' and result['score'] > best_score:
+            best_score = result['score']
+            best_intent = label
+    return best_intent
 
 # Agent 2: Domain Logic
 def handle_logic(intent):