Update pages/1_Detection.py

pages/1_Detection.py

@@ -20,15 +20,16 @@ st.session_state.selection = "" if "selection" not in st.session_state else st.s
 left, right = st.columns(2)
 selection = str(left.selectbox('Select document', os.listdir(MEDIA_ROOT)))
 st.session_state.predict = st.button('Detect')
+selection_image_left = MEDIA_ROOT+selection
+left.image(selection_image_left, use_column_width='always')
 
 if st.session_state.predict and st.session_state.selection:
     latest_detection = max(glob.glob(os.path.join(YOLO_RESULT, '*/')), key=os.path.getmtime)
     selection_detection =latest_detection + YOLO_OP + selection[:-4] + '.jpg'
-    st.image(selection_detection)
+    right.image(selection_detection)
 elif selection != st.session_state.selection:
     st.session_state.selection = selection
     detect.detect(MEDIA_ROOT)
-    selection_image_left = MEDIA_ROOT+selection
-    left.image(selection_image_left, use_column_width='always')
+