Update app.py

app.py

@@ -3,12 +3,14 @@ import transformers
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 import pandas as pd
+import seaborn as sns
+import matplotlib.pyplot as plt
 
 
 tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", padding= True, truncation=True, max_length=128)
 model = AutoModelForMaskedLM.from_pretrained("moussaKam/AraBART")
 
-
+#@st.cache
 def next_word(text, pipe):
     res_dict= {  
   'Word':[],
@@ -31,4 +33,11 @@ pipe = pipeline("fill-mask", tokenizer = tokenizer, model = model)
 dict_next_words = next_word(sent, pipe)
 df = pd.DataFrame.from_dict(dict_next_words)
 df.reset_index(drop=True, inplace=True)
-st.table(df)
+#design the graph theme
+sns.set_theme(style="whitegrid")
+sns.set_color_codes("pastel")
+
+fig = sns.barplot(data=df, orient = 'h')
+st.pyplot(fig)
+
+#st.table(df)