Update app.py

app.py

@@ -16,7 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained("moussaKam/AraBART", max_length=128, p
 model = AutoModelForMaskedLM.from_pretrained("Hamda/test-1-finetuned-AraBART")
 
 #@st.cache
-start = time()
+
 if (st.button('بحث', disabled=False)):
 
     def next_word(text, pipe):
@@ -36,10 +36,8 @@ if (st.button('بحث', disabled=False)):
     df = pd.DataFrame.from_dict(dict_next_words)
     df.reset_index(drop=True, inplace=True)
     st.dataframe(df)
-finish = time()  
-exec_time = 0
+#using Graph   
 if (st.checkbox('الاستعانة بالرسم البياني', value=False)):
-#if (st.button('استعمال الرسم البياني', disabled=False)):
     a = time()
     VocMap = './voc.csv'
     ScoreMap = './BM25.csv'
@@ -99,8 +97,7 @@ if (st.checkbox('الاستعانة بالرسم البياني', value=False)):
     res_df = pd.DataFrame.from_dict(dict_res)
     res_df.index += 1
     st.dataframe(res_df)
-b = time()
-exec_time = (b-a)
-exec_time_plm = (finish-start)   
-st.write(f'{str(exec_time)[:2] } :الوقت المستغرق باستعمال الرسم البياني')
+    b = time()
+    exec_time = (b-a)  
+    st.write(f'{str(exec_time)[:2] } :الوقت المستغرق باستعمال الرسم البياني')