Remove unused plotting function and clear button from sentiment analysis app

app.py

@@ -4,8 +4,6 @@
 from transformers import BertTokenizer, BertForSequenceClassification
 import torch
 import gradio as gr
-import torch.nn.functional as F
-import matplotlib.pyplot as plt
 
 # Load saved model and tokenizer
 model = BertForSequenceClassification.from_pretrained("./imdb_bert_model")
@@ -22,20 +20,6 @@ def predict_sentiment(text):
     sentiment = "Positive 😊" if prediction == 1 else "Negative 😠"
     return f"{sentiment} (Confidence: {confidence * 100:.2f}%)", probs.detach().numpy()[0]
 
-# Plotting function for probabilities
-def plot_probs(probs):
-    labels = ["Negative", "Positive"]
-    fig, ax = plt.subplots()
-    ax.bar(labels, probs, color=["red", "green"])
-    ax.set_ylim([0, 1])
-    ax.set_ylabel("Probability")
-    return fig
-
-# Clear button handler
-def clear_all():
-    return "", "", None
-
-
 # Responsive UI with gr.Blocks
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("## 🎬 IMDB Movie Review Sentiment Analyzer")
@@ -51,10 +35,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 autofocus=True
             )
             submit_btn = gr.Button("🔍 Analyze")
-            clear_btn = gr.Button("🧹 Clear")
         with gr.Column(scale=1):
             result_output = gr.Label(label="Predicted Sentiment")
-            prob_plot = gr.Plot(label="Confidence Scores")
 
     gr.Examples(
         examples=[
@@ -66,10 +48,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         inputs=[review_input]
     )
 
-    submit_btn.click(fn=predict_sentiment, inputs=review_input, outputs=[result_output,gr.Plot(label="Confidence Scores")])
-    clear_btn.click(fn=clear_all, outputs=[review_input, result_output, prob_plot])
-
+    submit_btn.click(fn=predict_sentiment, inputs=review_input, outputs=result_output)
     gr.Markdown("### Made with ❤️ by [Meet Mendapara](https://github.com/Meetmendapara09)")
 
-# Launch the app
 demo.launch(share=True)