sergioska commited on
Commit
e1b7276
·
1 Parent(s): b65bfeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -14,6 +14,7 @@ st.title('Upload an image file to detection')
14
 
15
  col1, col2, col3 = st.columns([0.5, 0.3, 0.2])
16
 
 
17
  with col1:
18
 
19
  st.header('flyer')
@@ -51,7 +52,15 @@ with col1:
51
  if crop_img.size > 0:
52
  cv2.imwrite("data/cropped_image_{}.jpg".format(i), crop_img)
53
  i += 1
54
-
 
55
  imageLocation.image(image)
56
  st.write(predictions)
57
 
 
 
 
 
 
 
 
 
14
 
15
  col1, col2, col3 = st.columns([0.5, 0.3, 0.2])
16
 
17
+ gibsOk = False
18
  with col1:
19
 
20
  st.header('flyer')
 
52
  if crop_img.size > 0:
53
  cv2.imwrite("data/cropped_image_{}.jpg".format(i), crop_img)
54
  i += 1
55
+
56
+ gibsOk = True
57
  imageLocation.image(image)
58
  st.write(predictions)
59
 
60
+ with col2:
61
+ st.header('gibs')
62
+ images = []
63
+ if gibsOk:
64
+ for cropped in glob.glob('data/*'):
65
+ st.write(cropped)
66
+