Update app.py

app.py

@@ -23,20 +23,21 @@ def next_word(text, pipe):
 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 += ' <mask>'
+# sent is the variable holding the user's input
+sent = st.text_area("Input", default_value, height=30)
+if len(sent)>20:
+   text_st = sent[-20:]
+   text_st += ' <mask>'
+else:
+   text_st = sent+ ' <mask>'
 
 pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model)
-dict_next_words = next_word(sent, pipe)
+dict_next_words = next_word(text_st, 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.dataframe(df)
 
 #st.bar_chart(df)