Spaces:
Runtime error
Runtime error
Edvin Behdadijd
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ model = YOLO('best.pt')
|
|
| 9 |
|
| 10 |
def detect_objects(image):
|
| 11 |
# Convert the input image to a format YOLO can work with
|
| 12 |
-
image = np.array(image)
|
| 13 |
|
| 14 |
# Perform detection
|
| 15 |
results = model(image)[0]
|
|
|
|
| 9 |
|
| 10 |
def detect_objects(image):
|
| 11 |
# Convert the input image to a format YOLO can work with
|
| 12 |
+
image = np.array(image,line_width=1)
|
| 13 |
|
| 14 |
# Perform detection
|
| 15 |
results = model(image)[0]
|