Update app.py

app.py

@@ -161,6 +161,25 @@ with col1:
     st.image(image, width=200)
 
 
+
+
+
+with col3:
+    st.markdown(f"##### YOLOX Standard vs SARAI Prediction:")
+    static_component = image_comparison(
+        img1=st.session_state["output_1"],
+        img2=st.session_state["output_2"],
+        label1="YOLOX",
+        label2="SARAI",
+        width=700,
+        starting_position=50,
+        show_labels=True,
+        make_responsive=True,
+        in_memory=True,
+    )
+
+    col1, col2, col3, col4, col5 = st.columns([1,2,3,2,2])
+with col1:
 # submit button
 submit = st.button("Perform Prediction")
 
@@ -194,21 +213,8 @@ if submit:
     st.session_state["output_1"] = output_1
     st.session_state["output_2"] = output_2
 
-
-
-with col3:
-    st.markdown(f"##### YOLOX Standard vs SARAI Prediction:")
-    static_component = image_comparison(
-        img1=st.session_state["output_1"],
-        img2=st.session_state["output_2"],
-        label1="YOLOX",
-        label2="SARAI",
-        width=700,
-        starting_position=50,
-        show_labels=True,
-        make_responsive=True,
-        in_memory=True,
-    )
+    
+with col4:
     st.markdown(f"##### Slide to Compare")