Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ import glob
|
|
24 |
def predict(text,img):
|
25 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
|
26 |
img = cv2.imread(img)
|
27 |
-
results = model(
|
28 |
dirpath = tempfile.mkdtemp()
|
29 |
results.crop(save_dir=dirpath)
|
30 |
path= dirpath+'/crops/**/*.jpg'
|
|
|
24 |
def predict(text,img):
|
25 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
|
26 |
img = cv2.imread(img)
|
27 |
+
results = model(img)
|
28 |
dirpath = tempfile.mkdtemp()
|
29 |
results.crop(save_dir=dirpath)
|
30 |
path= dirpath+'/crops/**/*.jpg'
|