Tweak formatting

app.py

@@ -56,7 +56,7 @@ st.markdown("# DocQuery: Query Documents w/ NLP")
 if "document" not in st.session_state:
     st.session_state["document"] = None
 
-input_col, model_col = st.columns(2)
+input_col, model_col = st.columns([2,1])
 
 with input_col:
     input_type = st.radio("Pick an input type", ["Upload", "URL"], horizontal=True)
@@ -103,7 +103,7 @@ question = st.text_input("QUESTION", "")
 document = st.session_state.document
 loading_placeholder = st.empty()
 if document is not None:
-    col1, col2 = st.columns([3, 1])
+    col1, col2 = st.columns(2)
     image = document.preview
 
 
@@ -129,7 +129,7 @@ if document is not None and question is not None and len(question) > 0:
                 image.width,
                 image.height,
             )
-            draw.rectangle(((x1, y1), (x2, y2)), outline=colors[i])
+            draw.rectangle(((x1, y1), (x2, y2)), outline=colors[i], width=3)
 
 if document is not None:
     col1.image(image, use_column_width='auto')