cosmetics

lfoppiano · Jun 24, 2024 · 6eee84d · 6eee84d
1 parent d428544
commit 6eee84d
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/streamlit_app.py b/streamlit_app.py
@@ -425,12 +425,13 @@ def generate_color_gradient(num_elements):
 
         text_response = None
         if mode == "embeddings":
-            with st.spinner("Fetching the relevant context..."):
-                text_response, coordinates = st.session_state['rqa'][model].query_storage(
-                    question,
-                    st.session_state.doc_id,
-                    context_size=context_size
-                )
+            with placeholder:
+                with st.spinner("Fetching the relevant context..."):
+                    text_response, coordinates = st.session_state['rqa'][model].query_storage(
+                        question,
+                        st.session_state.doc_id,
+                        context_size=context_size
+                    )
         elif mode == "llm":
             with placeholder:
                 with st.spinner("Generating LLM response..."):