Update app.py

app.py

@@ -19,13 +19,13 @@ model = AutoModelForMaskedLM.from_pretrained("Hamda/test-1-finetuned-AraBART")
 if (st.button('بحث', disabled=False)):
     def next_word(text, pipe):
         res_dict= {  
-          'Word':[],
-          'Score':[],
+          'الكلمة المقترحة':[],
+          'العلامة':[],
         }
         for e in pipe(text):
             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'])
+                res_dict['الكلمة المقترحة'].append(e['token_str'])
+                res_dict['العلامة'].append(e['score'])
         return res_dict
     
     text_st = sent+ ' <mask>'
@@ -79,8 +79,8 @@ if (st.button('استعمال الرسم البياني', disabled=False)):
     dict_res = {'word':[], 'score':[]}
     for key, value in tmexp:
         new_score=((value-min(t_li))/(max(t_li)-min(t_li)))-0.0001
-        dict_res['score'].append(str(new_score)[:6])
-        dict_res['word'].append(key)
+        dict_res['العلامة'].append(str(new_score)[:6])
+        dict_res['الكلمة المقترحة'].append(key)
         i+=1
         if (i==10):
             break