Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import pickle
|
|
| 8 |
# function which is returning the number of object detected
|
| 9 |
def number_object_detected(image):
|
| 10 |
|
| 11 |
-
custom_model = YOLO('
|
| 12 |
results = custom_model(image,verbose= False)
|
| 13 |
|
| 14 |
dic = results[0].names
|
|
|
|
| 8 |
# function which is returning the number of object detected
|
| 9 |
def number_object_detected(image):
|
| 10 |
|
| 11 |
+
custom_model = YOLO('best2.pt') # custome yolo model path
|
| 12 |
results = custom_model(image,verbose= False)
|
| 13 |
|
| 14 |
dic = results[0].names
|