Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import transformers
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForMaskedLM
+import pandas as pd
+
 
 tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", padding= True, truncation=True, max_length=128)
 model = AutoModelForMaskedLM.from_pretrained("moussaKam/AraBART")
@@ -27,5 +29,5 @@ sent += ' <mask>'
 
 pipe = pipeline("fill-mask", tokenizer = tokenizer, model = model)
 dict_next_words = next_word(sent, pipe)
-
-st.write(dict_next_words)
+df = pd.DataFrame.from_dict(dict_next_words)
+st.write(df)