Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
+import pandas as pd
+import matplotlib.pyplot as plt
+import seaborn as sns
 import transformers
-import torch
 
 # Define sentiment analysis models
 models = {
@@ -32,9 +34,15 @@ def app():
                 results.append((model_name, label, score))
         st.success("Sentiment analysis complete!")
         st.write("Results:")
-        for model_name, label, score in results:
-            st.write(f"- {model_name}: {label} ({score:.2f})")
+        df = pd.DataFrame(results, columns=["Model", "Sentiment", "Score"])
+        st.write(df)
 
+        # Plot results
+        sns.set_style("whitegrid")
+        fig, ax = plt.subplots()
+        sns.barplot(x="Model", y="Score", hue="Sentiment", data=df, ax=ax)
+        ax.set_title("Sentiment Analysis Results")
+        st.pyplot(fig)
 
 # Run Streamlit app
 if __name__ == "__main__":