Update pages/type_text.py

pages/type_text.py

@@ -159,6 +159,11 @@ 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)
+
 if INTdesc_input is not None and st.button("Map to SBS codes"): 
     HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
     HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
@@ -185,11 +190,6 @@ if INTdesc_input is not None and st.button("Map to SBS codes"):
     prompt = question + " " +"A: "+ shortlist[0] + " " +"B: " + shortlist[1] + " " + "C: " + shortlist[2] + " " + "D: " + shortlist[3] + " " + "E: " + shortlist[4]
     #st.write(prompt)
 
-    sp = iter(make_spinner()) # this starts the spinning
-    next(sp) # heavy computation
-    time.sleep(3) # spinning end
-    next(sp, None)
-
     messages = [
     {"role": "system", "content": "You are a knowledgable AI assistant who always answers truthfully and precisely!"},
     {"role": "user", "content": prompt},