Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def yolov8_img_inference(
|
|
50 |
# get the model names list
|
51 |
names = model.names
|
52 |
# get the 'obj' class id
|
53 |
-
obj_id = list(names)[list(names.values()).index('hop_dln')
|
54 |
# ,'hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
|
55 |
# obj_id = list(names)[list(names.values()).index([0])]
|
56 |
# count 'car' objects in the results
|
|
|
50 |
# get the model names list
|
51 |
names = model.names
|
52 |
# get the 'obj' class id
|
53 |
+
obj_id = list(names)[list(names.values()).index('hop_dln')]
|
54 |
# ,'hop_jn','hop_vtg','hop_ytv','lo_kids', 'lo_ytv','loc_dln','loc_jn','loc_kids','loc_ytv')]
|
55 |
# obj_id = list(names)[list(names.values()).index([0])]
|
56 |
# count 'car' objects in the results
|