Update app.py

app.py

@@ -20,7 +20,7 @@ def next_word(text, pipe):
     
   
     for e in pipe(text):
-        if e['token_str'] not in string.punctuation:
+        if not any(e['token_str'] in string.punctuation):
             res_dict['Word'].append(e['token_str'])
             res_dict['Score'].append(e['score'])
     return res_dict
@@ -34,7 +34,7 @@ st.checkbox('استعمال الرسم البياني', value=False)
 
 text_st = sent+ ' <mask>'
 
-pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model, top_k=10)
+pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model, top_k=6)
 dict_next_words = next_word(text_st, pipe)
 df = pd.DataFrame.from_dict(dict_next_words)
 df.reset_index(drop=True, inplace=True)