Update pages/type_text.py

pages/type_text.py

@@ -5,6 +5,7 @@ import json
 import torch
 from transformers import pipeline # AutoTokenizer, AutoModelForCausalLM, AutoModelForTokenClassification
 from sentence_transformers import SentenceTransformer, util 
+import time
 
 import os
 os.getenv("HF_TOKEN") 
@@ -73,7 +74,11 @@ def get_device_map() -> str:
 device = get_device_map()  # 'cpu'
 
 def on_click():
-    st.session_state.user_input = ""
+    st.session_state.user_input = "" 
+
+def make_spinner(text = "In progress..."):
+    with st.spinner(text):
+        yield
     
 #@st.cache
 def convert_df(df:pd.DataFrame):
@@ -180,9 +185,11 @@ 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},