Add loading placeholder

app.py

@@ -56,8 +56,13 @@ st.markdown("# DocQuery: Query Documents w/ NLP")
 if "document" not in st.session_state:
     st.session_state["document"] = None
 
-input_type = st.radio("Pick an input type", ["Upload", "URL"], horizontal=True)
+input_col, model_col = st.columns(2)
 
+with input_col:
+    input_type = st.radio("Pick an input type", ["Upload", "URL"], horizontal=True)
+
+with model_col:
+    model_type = st.radio("Pick a model", ["Upload", "URL"], horizontal=True)
 
 def load_file_cb():
     if st.session_state.file_input is None:
@@ -105,20 +110,26 @@ if document is not None:
 colors = ["blue", "red", "green"]
 if document is not None and question is not None and len(question) > 0:
     col2.header("Answers")
-
-    predictions = run_pipeline(question=question, document=document, top_k=1)
-
-    word_boxes = lift_word_boxes(document)
-    image = image.copy()
-    draw = ImageDraw.Draw(image)
-    for i, p in enumerate(ensure_list(predictions)):
-        col2.markdown(f"#### { p['answer'] }: ({round(p['score'] * 100, 1)}%)")
-        x1, y1, x2, y2 = normalize_bbox(
-            expand_bbox(word_boxes[p["start"] : p["end"] + 1]),
-            image.width,
-            image.height,
-        )
-        draw.rectangle(((x1, y1), (x2, y2)), outline=colors[i])
+    with col2:
+        answers_placeholder = st.empty()
+        answers_loading_placeholder = st.empty()
+
+    with answers_loading_placeholder:
+        with st.spinner("Processing question..."):
+            predictions = run_pipeline(question=question, document=document, top_k=1)
+
+    with answers_placeholder:
+        word_boxes = lift_word_boxes(document)
+        image = image.copy()
+        draw = ImageDraw.Draw(image)
+        for i, p in enumerate(ensure_list(predictions)):
+            col2.markdown(f"#### { p['answer'] }: ({round(p['score'] * 100, 1)}%)")
+            x1, y1, x2, y2 = normalize_bbox(
+                expand_bbox(word_boxes[p["start"] : p["end"] + 1]),
+                image.width,
+                image.height,
+            )
+            draw.rectangle(((x1, y1), (x2, y2)), outline=colors[i])
 
 if document is not None:
     col1.image(image, use_column_width='auto')