Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TextClassificationPipeline
 import operator
-import plotly.express as px
+import matplotlib.pyplot as plt
 
 def get_sentiment(out):
   d = dict()
@@ -28,7 +28,8 @@ text = st.text_area(f'Ciao! This app uses {model_name}.\nEnter your text  to tes
 if text:
   out = pipe(text)
   df = get_sentiment(out[0])
-  fig = px.pie(df, values='score', names='sentiment')
+  fig, ax = plt.subplots()
+  ax.bar(df.index, df[0])
   
   st.pyplot(fig)