Update app.py

app.py

@@ -15,8 +15,7 @@ def next_word(text, pipe):
   'Word':[],
   'Score':[],
   }
-    res=pipe(text)
-    for e in res:
+    for e in pipe(text):
         res_dict['Word'].append(e['token_str'])
         res_dict['Score'].append(e['score'])
     return res_dict
@@ -25,15 +24,21 @@ st.title("Predict Next Word")
 st.write("Expand your query by leveraging various models")
 default_value = "التاريخ هو تحليل و"
 # sent is the the variable holding the user's input
-sent = st.text_area("Text", default_value, height = 30)
+sent = st.text_area("Text", default_value, height=30)
 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)
 
+chart_data = pd.DataFrame(
+     np.random.randn(50, 3),
+     columns=["a", "b", "c"])
+
+st.bar_chart(chart_data)
+#st.dataframe(df)
+
+#st.bar_chart(df)
 
-st.bar_chart(df)
-#st.dataframe(df.style.highlight_max(axis=0))
 #st.table(df)