Update app.py

app.py

@@ -11,17 +11,16 @@ def zeroShotClassification(text_input, candidate_labels):
     # Get predictions
     prediction = classifier(text_input, labels)
     
-    # Format results as percentage with 2 decimal places
-    results = {label: f"{score*100:.2f}%" 
-              for label, score in zip(prediction['labels'], prediction['scores'])}
+    # For Label component: Return raw scores (not percentage strings)
+    results = {label: score for label, score in zip(prediction['labels'], prediction['scores'])}
     
-    # Create markdown output for detailed view
+    # For Markdown: Create detailed view with percentage formatting
     markdown_output = "### Results Breakdown:\n\n"
-    for label, score in sorted(results.items(), key=lambda x: float(x[1].rstrip('%')), reverse=True):
-        # Create confidence bar using Unicode blocks
-        score_num = float(score.rstrip('%'))
-        blocks = "█" * int(score_num/5) + "░" * (20 - int(score_num/5))
-        markdown_output += f"**{label}**: {blocks} {score}\n\n"
+    for label, score in sorted(results.items(), key=lambda x: x[1], reverse=True):
+        # Convert to percentage for display
+        percentage = score * 100
+        blocks = "█" * int(percentage/5) + "░" * (20 - int(percentage/5))
+        markdown_output += f"**{label}**: {blocks} {percentage:.2f}%\n\n"
     
     return results, markdown_output
 
@@ -66,7 +65,7 @@ demo = gr.Interface(
         ["The team scored in the final minute to win the championship", "sports, victory, competition, excitement"],
         ["The painting uses vibrant colors to express deep emotions", "art, emotion, creativity, analysis"]
     ],
-    cache_examples=False,  # Disable caching to prevent the error
+    cache_examples=False,
     css="""
     footer {display:none !important}
     .output-markdown{display:none !important}
@@ -113,8 +112,7 @@ demo = gr.Interface(
         font-family: 'Inter', sans-serif !important;
         color: #11142D !important;
     }
-    """,
-    theme=gr.themes.Soft()
+    """
 )
 
 # Launch the app