Update app.py

app.py

@@ -16,9 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", max_length=128, p
 model = AutoModelForMaskedLM.from_pretrained("Hamda/test-1-finetuned-AraBART")
 pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model, top_k=10)
 #@st.cache
-if (st.button('بحث', disabled=False)):
-
-    def next_word(text, pipe):
+def next_word(text, pipe):
         res_dict= {  
           'الكلمة المقترحة':[],
           'العلامة':[],
@@ -28,7 +26,7 @@ if (st.button('بحث', disabled=False)):
                 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)