Update app.py

app.py

@@ -77,11 +77,11 @@ def detect_context(input_text, threshold=0.03):
     label_probabilities = list(zip(labels, probabilities))
 
     # Filter contexts with confidence >= threshold
-    high_confidence_contexts = [(label, score) for label, score in label_probabilities if score >= threshold]
+    high_confidence_contexts = [label in label_probabilities if score >= threshold]
 
     # If no contexts meet the threshold, default to "general"
     if not high_confidence_contexts:
-        high_confidence_contexts = [("general", 1.0)]  # Assign a default score of 1.0 for "general"
+        high_confidence_contexts = [("general")]
 
     return high_confidence_contexts