Update app.py

app.py

@@ -110,7 +110,6 @@ A Demonstration of SARAI's Utility
     unsafe_allow_html=True,
 )
 
-
 st.write("##")
 
 with st.expander("Instructions for Use"):
@@ -131,9 +130,7 @@ with st.expander("Instructions for Use"):
 
 st.write("##")
 
-
-
-col1, col2 = st.columns([3,6])
+col1, col2, col3 = st.columns([3, 1, 6])
 with col1:
     st.markdown(f"##### Set input image:")
 
@@ -166,19 +163,19 @@ with col1:
         image = sahi.utils.cv.read_image_as_pil(IMAGE_TO_URL[slider])
     st.image(image, width=300)
 
-with col2:
+with col3:
     st.markdown(f"##### YOLOX Standard vs SAHI Prediction:")
-static_component = image_comparison(
-    img1=st.session_state["output_1"],
-    img2=st.session_state["output_2"],
-    label1="YOLOX",
-    label2="SAHI+YOLOX",
-    width=700,
-    starting_position=50,
-    show_labels=True,
-    make_responsive=True,
-    in_memory=True,
-)
+    static_component = image_comparison(
+        img1=st.session_state["output_1"],
+        img2=st.session_state["output_2"],
+        label1="YOLOX",
+        label2="SAHI+YOLOX",
+        width=700,
+        starting_position=50,
+        show_labels=True,
+        make_responsive=True,
+        in_memory=True,
+    )
 
 col1, col2, col3 = st.columns([4, 3, 4])
 with col2: