Update app.py

app.py

@@ -22,11 +22,11 @@ def classify_text(text, candidate_labels):
         candidate_labels (str): Comma-separated string of possible labels
     
     Returns:
-        list: List of (label, score) tuples
+        dict: Dictionary with labels as keys and confidence scores as values
     """
     if classifier is None:
-        # Return a default response when model fails to load
-        return [("error", 1.0)]
+        # Return default response when model fails to load
+        return {"error": 1.0}
     
     try:
         # Convert comma-separated string to list
@@ -35,12 +35,12 @@ def classify_text(text, candidate_labels):
         # Perform classification
         result = classifier(text, labels)
         
-        # Convert results to list of (label, score) tuples
-        return list(zip(result["labels"], result["scores"]))
+        # Convert to dictionary format that Gradio Label expects
+        return {label: float(score) for label, score in zip(result["labels"], result["scores"])}
     
     except Exception as e:
         print(f"Classification error: {e}")
-        return [("error", 1.0)]
+        return {"error": 1.0}
 
 # Create Gradio interface
 iface = gr.Interface(