Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import shutil
|
|
8 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
|
9 |
|
10 |
def detect_image(image):
|
11 |
-
results =
|
12 |
return results.render()[0]
|
13 |
|
14 |
|
|
|
8 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')
|
9 |
|
10 |
def detect_image(image):
|
11 |
+
results = model(image)
|
12 |
return results.render()[0]
|
13 |
|
14 |
|