Update app.py

app.py

@@ -12,9 +12,15 @@ def predict(text):
     label = result["label"]
     score = round(result["score"], 4)
 
-    sentiment_emoji = "😊" if label == "POSITIVE" else "😡"
+    # Map labels properly
+    if label == "LABEL_1":
+        sentiment = "Positive"
+        emoji = "😊"
+    else:
+        sentiment = "Negative"
+        emoji = "😡"
 
-    return f"{sentiment_emoji} {label}", f"Confidence: {score}", score
+    return f"{emoji} {sentiment}", f"Confidence: {score}", score
 
 
 with gr.Blocks(theme=gr.themes.Soft(), title="IMDb Sentiment Analyzer") as demo: