Update app.py

app.py

@@ -163,6 +163,20 @@ with col1:
         image = sahi.utils.cv.read_image_as_pil(IMAGE_TO_URL[slider])
     st.image(image, width=200)
 
+# submit button
+submit = st.button("🐦 Perform Prediction")
+
+if submit:
+    # perform prediction
+    with st.spinner(
+        text="Downloading model weight.. "
+        + st.session_state["last_spinner_texts"].get()
+    ):
+        detection_model = get_model()
+
+    image_size = 416
+
+
 with col3:
     st.markdown(f"##### YOLOX Standard vs SAHI Prediction:")
     static_component = image_comparison(
@@ -177,19 +191,7 @@ with col3:
         in_memory=True,
     )
 
-col1, col2, col3 = st.columns([4, 3, 4])
-with col2:
-    submit = st.button("🐦 Perform Prediction")
-
-if submit:
-    # perform prediction
-    with st.spinner(
-        text="Downloading model weight.. "
-        + st.session_state["last_spinner_texts"].get()
-    ):
-        detection_model = get_model()
 
-    image_size = 416
 
     with st.spinner(
         text="Performing prediction.. " + st.session_state["last_spinner_texts"].get()