SciStalk commited on
Commit
3d795d9
1 Parent(s): d6a1bbc
Files changed (2) hide show
  1. app.py +3 -1
  2. example_images/empty.txt +0 -0
app.py CHANGED
@@ -4,7 +4,9 @@ import cv2
4
  import numpy as np
5
  from ultralytics import YOLO
6
 
7
- image_directory = 'home/user/app/example_images'
 
 
8
 
9
  img_files = [file for file in os.listdir(
10
  image_directory) if file.lower().endswith('.jpg') or file.lower().endswith('.png')]
 
4
  import numpy as np
5
  from ultralytics import YOLO
6
 
7
+ image_directory = '/home/user/app/example_images'
8
+
9
+ os.makedirs(image_directory, exist_ok=True)
10
 
11
  img_files = [file for file in os.listdir(
12
  image_directory) if file.lower().endswith('.jpg') or file.lower().endswith('.png')]
example_images/empty.txt ADDED
File without changes