Update pages/type_text.py

pages/type_text.py

@@ -159,10 +159,10 @@ pipe = load_pipe()
 INTdesc_embedding = model.encode(INTdesc_input)
 SBScorpus_embeddings = model.encode(SBScorpus)
 
-sp = iter(make_spinner()) # this starts the spinning
-next(sp) # heavy computation
-time.sleep(3) # spinning end
-next(sp, None)
+#sp = iter(make_spinner()) # this starts the spinning
+#next(sp) # heavy computation
+#time.sleep(3) # spinning end
+#next(sp, None)
 
 if INTdesc_input is not None and st.button("Map to SBS codes"): 
     HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
@@ -198,8 +198,13 @@ if INTdesc_input is not None and st.button("Map to SBS codes"):
         messages,
         max_new_tokens=256,
     )
-    st.write(outputs[0]["generated_text"][-1]["content"])
-    
+    st.write(outputs[0]["generated_text"][-1]["content"]) 
+
+    sp = iter(make_spinner()) # this starts the spinning
+    next(sp) # heavy computation
+    time.sleep(15) # spinning end
+    next(sp, None)
+
     bs, b1, b2, b3, bLast = st.columns([0.75, 1.5, 1.5, 1.5, 0.75])
     with b1:
         #csvbutton = download_button(results, "results.csv", "📥 Download .csv")