Update app.py

app.py

@@ -33,20 +33,15 @@ def next_word(text, pipe):
             res_dict['الكلمة المقترحة'].append(e['token_str'])
             res_dict['العلامة'].append(e['score'])
     return res_dict
-
 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)
     df.reset_index(drop=True, inplace=True)
     st.dataframe(df)
 #using Graph   
 if (st.checkbox('الاستعانة بالرسم البياني المعرفي الاحتمالي', value=False)):
     a = time()
-    state_cb = True
     VocMap = './voc.csv'
     ScoreMap = './BM25.csv'
     
@@ -95,10 +90,6 @@ if (st.checkbox('الاستعانة بالرسم البياني المعرفي
         return tmt
     
     tmt = compute_score(L, id_list)    
-    
-    for e in sent:
-        del tmt[e]
-    
     exp_terms = []
     t_li = tmt.values()
     tmexp = sorted(tmt.items(), key=lambda x: x[1], reverse=True)