Update app.py

app.py

@@ -23,16 +23,35 @@ def predict(text):
     predicted_class = torch.argmax(probs, dim=1).item()
     confidence = probs[0][predicted_class].item()
 
-    label = "Positive 😊" if predicted_class == 1 else "Negative 😡"
-
-    return f"{label} (Confidence: {confidence:.2f})"
-
-demo = gr.Interface(
-    fn=predict,
-    inputs=gr.Textbox(lines=3, placeholder="Enter text..."),
-    outputs="text",
-    title="🎬 Sentiment Analyzer",
-    description="BERT-based sentiment classifier"
-)
+    label = "🟢 Positive" if predicted_class == 1 else "🔴 Negative"
+
+    return label, f"{confidence:.2f}"
+
+# 🎨 CUSTOM UI
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    
+    gr.Markdown("""
+    # 🎬 AI Sentiment Analyzer
+    ### Analyze emotions in text using BERT 🤖
+    """)
+    
+    with gr.Row():
+        text_input = gr.Textbox(
+            placeholder="Type your sentence here...",
+            lines=3,
+            label="Input Text"
+        )
+
+    analyze_btn = gr.Button("Analyze Sentiment 🚀")
+
+    with gr.Row():
+        result_label = gr.Textbox(label="Prediction")
+        confidence_score = gr.Textbox(label="Confidence")
+
+    analyze_btn.click(
+        fn=predict,
+        inputs=text_input,
+        outputs=[result_label, confidence_score]
+    )
 
 demo.launch()