sergioska commited on
Commit
4bf5062
·
1 Parent(s): ed79734

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import cv2
2
  import glob
3
  import torch
@@ -11,6 +12,9 @@ from st_clickable_images import clickable_images
11
 
12
  st.set_page_config(layout="wide")
13
 
 
 
 
14
  detector = pipeline("object-detection", model="sergioska/yolo-finetuned-flyers")
15
 
16
  st.title('Upload an image file to detection')
 
1
+ import os
2
  import cv2
3
  import glob
4
  import torch
 
12
 
13
  st.set_page_config(layout="wide")
14
 
15
+ for filename in glob.glob('/tmp/cropped_image_*'):
16
+ os.remove(filename)
17
+
18
  detector = pipeline("object-detection", model="sergioska/yolo-finetuned-flyers")
19
 
20
  st.title('Upload an image file to detection')