Update app.py

app.py

@@ -3,8 +3,7 @@ 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)
@@ -29,14 +28,9 @@ default_value = "التاريخ هو تحليل و"
 sent = st.text_area("Text", default_value, height = 30)
 sent += ' <mask>'
 
-pipe = pipeline("fill-mask", tokenizer = tokenizer, model = model)
+pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model)
 dict_next_words = next_word(sent, pipe)
 df = pd.DataFrame.from_dict(dict_next_words)
-#design the graph theme
-sns.set_theme(style="whitegrid")
-sns.set_color_codes("pastel")
-fig, ax = plt.subplots(figsize=(6, 15))
-sns.barplot(x='Score', y='Word', data=df, orient = 'h')
-st.pyplot(fig)
+df.reset_index(drop=True, inplace=True)
 
-#st.table(df)
+st.table(df)