Update app.py

app.py

@@ -38,8 +38,7 @@ if (st.button('بحث', disabled=False)):
     text_st = sent+ ' <mask>'
     dict_next_words = next_word(text_st, pipe)
     df = pd.DataFrame.from_dict(dict_next_words)
-    new_df = df[~df['الكلمة المقترحة'].isin(sent)]
-    st.dataframe(new_df)
+    st.dataframe(df)
 #using Graph   
 
 if (st.checkbox('الاستعانة بالرسم البياني المعرفي الاحتمالي', value=False)):
@@ -116,15 +115,13 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
     exec_time = (b-a)
     text_st = sent+ ' <mask>'
     dict_next_words = next_word(text_st, pipe)
-    
     df = pd.DataFrame.from_dict(dict_next_words)
-    new_df = df[~df['الكلمة المقترحة'].isin(sent)]
-    new_df.index += 1
+    df.index += 1
     str_time = str(exec_time)[:3]
     
     st.markdown("""---""")
     st.header("الكلمات المقترحة باستعمال النموذج اللغوي")
-    st.dataframe(new_df)
+    st.dataframe(df)
     st.markdown("""---""")
     st.header("الكلمات المقترحة باستعمال الرسم البياني")
     st.dataframe(res_df)