improve checking_model output UI

containers · Feb 14, 2024 · 0c1e548 · 0c1e548
1 parent 56c4da5
commit 0c1e548
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/chatbot-langchain/chatbot_ui.py b/chatbot-langchain/chatbot_ui.py
@@ -10,8 +10,9 @@
 model_service = os.getenv("MODEL_SERVICE_ENDPOINT",
                           "http://localhost:8001/v1")
 
-@st.cache_resource
+@st.cache_resource(show_spinner=False)
 def checking_model_service():
+    start = time.time()
     print("Checking Model Service Availability...")
     ready = False
     while not ready:
@@ -23,8 +24,10 @@ def checking_model_service():
             pass
         time.sleep(1) 
     print("Model Service Available")
+    print(f"{time.time()-start} seconds")
 
-checking_model_service()
+with st.spinner("Checking Model Service Availability..."):
+    checking_model_service()
 
 st.title("💬 Chatbot")  
 if "messages" not in st.session_state: