MarkusEssl commited on
Commit
172e9ef
·
1 Parent(s): 593cc62
Files changed (1) hide show
  1. pages/1_Detection.py +1 -1
pages/1_Detection.py CHANGED
@@ -27,5 +27,5 @@ st.session_state.predict = st.button('Detect')
27
 
28
  if st.session_state.predict:
29
  latest_detection = max(glob.glob(os.path.join(YOLO_RESULT, '*/')), key=os.path.getmtime)
30
- selection_detection =latest_detection + YOLO_OP + selection + '.jpg'
31
  st.image(selection_detection)
 
27
 
28
  if st.session_state.predict:
29
  latest_detection = max(glob.glob(os.path.join(YOLO_RESULT, '*/')), key=os.path.getmtime)
30
+ selection_detection =latest_detection + YOLO_OP + selection[:-4] + '.jpg'
31
  st.image(selection_detection)