Update pages/type_text.py

pages/type_text.py

@@ -175,9 +175,6 @@ SBScorpus_embeddings = model.encode(SBScorpus)
 #    my_bar.progress(p.progress, text=f"Progress: {p.progress}%")
 #my_bar.empty()
 
-for _ in stqdm(range(50), desc="This is a slow task", mininterval=1):
-    time.sleep(0.5)  
-
 
 #progress_bar = st.progress(0)
 #status_text = st.empty()
@@ -185,7 +182,8 @@ for _ in stqdm(range(50), desc="This is a slow task", mininterval=1):
 #    #time.sleep(0.5)
 #    progress_bar.progress(pct_complete)
 #    status_text.write("Progress: {}/100".format(pct_complete))
-##status_text.write("Progress: {}".format("Done"))
+##status_text.write("Progress: {}".format("Done")) 
+#progress_bar.empty()
 
 if INTdesc_input is not None and st.button("Map to SBS codes", key="run_st_model"): 
     HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
@@ -218,6 +216,14 @@ if INTdesc_input is not None and st.button("Map to SBS codes", key="run_st_model
     {"role": "user", "content": prompt},
     ]
 
+    progress_bar = st.progress(0)
+    status_text = st.empty()
+    for pct_complete in range(1,101):
+        time.sleep(0.5)
+        progress_bar.progress(pct_complete)
+        #status_text.write("Progress: {}/100".format(pct_complete))
+        progress_bar.empty()
+    
     outputs = pipe(
         messages,
         max_new_tokens=256,