Update app.py

app.py

@@ -14,9 +14,8 @@ sent = st.text_area('مدخل',default_value)
 
 tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", max_length=128, padding=True, pad_to_max_length = True, truncation=True)
 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):
@@ -31,7 +30,6 @@ if (st.button('بحث', disabled=False)):
         return res_dict
     
     text_st = sent+ ' <mask>'
-    pipe = pipeline("fill-mask", tokenizer=tokenizer, model=model, top_k=10)
     dict_next_words = next_word(text_st, pipe)
     df = pd.DataFrame.from_dict(dict_next_words)
     df.reset_index(drop=True, inplace=True)
@@ -96,8 +94,16 @@ if (st.checkbox('الاستعانة بالرسم البياني', value=False)):
             break
     res_df = pd.DataFrame.from_dict(dict_res)
     res_df.index += 1
+    if (len())
     st.dataframe(res_df)
     b = time()
-    exec_time = (b-a)  
+    exec_time = (b-a)
+    
+    text_st = sent+ ' <mask>'
+    dict_next_words = next_word(text_st, pipe)
+    df = pd.DataFrame.from_dict(dict_next_words)
+    df.reset_index(drop=True, inplace=True)
+    st.dataframe(df)
+    
     st.write(f'{str(exec_time)[:2] } :الوقت المستغرق باستعمال الرسم البياني')