Update app.py

app.py

@@ -31,7 +31,9 @@ sent += ' <mask>'
 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)
+#df.reset_index(drop=True, inplace=True)
 
-st.dataframe(df.style.highlight_max(axis=0))
+st.bar_chart(data=df, use_container_width=True)
+
+#st.dataframe(df.style.highlight_max(axis=0))
 #st.table(df)