Update app.py

app.py

@@ -24,14 +24,12 @@ model = AutoModelForMaskedLM.from_pretrained("Hamda/test-1-finetuned-AraBART")
 pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model, top_k=10)
 
 def next_word(text, pipe):
-    
-    filter_words = list(string.punctuation).extend(sent)
     res_dict= {  
       'الكلمة المقترحة':[],
       'العلامة':[],
     }
     for e in pipe(text):
-        if all(c not in filter_words for c in e['token_str']):
+        if all(c not in list(string.punctuation) for c in e['token_str']):
             res_dict['الكلمة المقترحة'].append(e['token_str'])
             res_dict['العلامة'].append(e['score'])
     return res_dict