Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,7 @@ import glob
|
|
22 |
# Model
|
23 |
|
24 |
def predict(img,text):
|
|
|
25 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
|
26 |
results = model(img)
|
27 |
dirpath = tempfile.mkdtemp()
|
|
|
22 |
# Model
|
23 |
|
24 |
def predict(img,text):
|
25 |
+
import tempfile
|
26 |
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
|
27 |
results = model(img)
|
28 |
dirpath = tempfile.mkdtemp()
|