add chart

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-
+import plotly.express as px
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 txt = st.text_area('Text to analyze', '''
     It was the best of times, it was the worst of times, it was
@@ -19,4 +19,7 @@ model = AutoModelForSequenceClassification.from_pretrained("s-nlp/roberta_toxici
 batch = tokenizer.encode('txt', return_tensors='pt')
 
 # inference
-model(batch)
+result = model(batch)
+
+fig = px.bar(result, x="", y="", orientation='h')
+fig.show()

requirements.txt

@@ -1,2 +1,3 @@
 transformers
-torch
+torch
+plotly