Update app.py

app.py

@@ -9,7 +9,7 @@ import pandas as pd
 tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", padding= True, truncation=True, max_length=128)
 model = AutoModelForMaskedLM.from_pretrained("moussaKam/AraBART")
 
-#@st.cache
+@st.cache
 def next_word(text, pipe):
     res_dict= {  
   'Word':[],
@@ -33,4 +33,5 @@ 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)
+st.dataframe(df.style.highlight_max(axis=0))
+#st.table(df)