Update app.py

app.py

@@ -13,14 +13,11 @@ model = AutoModelForMaskedLM.from_pretrained("Hamda/test-1-finetuned-AraBART")
 #@st.cache
 def next_word(text, pipe):
     res_dict= {  
-  'Word':[],
-  'Score':[],
-  }
-   
-    
-  
+      'Word':[],
+      'Score':[],
+    }
     for e in pipe(text):
-        if not any(e['token_str'] in string.punctuation):
+        if all(c not in list(string.punctuation) for c in e['token_str']):
             res_dict['Word'].append(e['token_str'])
             res_dict['Score'].append(e['score'])
     return res_dict