Update app.py

app.py

@@ -22,10 +22,11 @@ def classify_text(text, candidate_labels):
         candidate_labels (str): Comma-separated string of possible labels
     
     Returns:
-        dict: Dictionary containing labels and their corresponding scores
+        list: List of (label, score) tuples
     """
     if classifier is None:
-        return {"Error": "Model failed to load"}
+        # Return a default response when model fails to load
+        return [("error", 1.0)]
     
     try:
         # Convert comma-separated string to list
@@ -34,15 +35,12 @@ def classify_text(text, candidate_labels):
         # Perform classification
         result = classifier(text, labels)
         
-        # Create formatted output
-        output = {}
-        for label, score in zip(result["labels"], result["scores"]):
-            output[label] = f"{score:.4f}"
-            
-        return output
+        # Convert results to list of (label, score) tuples
+        return list(zip(result["labels"], result["scores"]))
     
     except Exception as e:
-        return {"Error": str(e)}
+        print(f"Classification error: {e}")
+        return [("error", 1.0)]
 
 # Create Gradio interface
 iface = gr.Interface(