linhcuem commited on
Commit
39cc8c5
1 Parent(s): 1682426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -50,8 +50,8 @@ 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
57
  # count_result = results[0].boxes.cls[0].item()
 
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('lo_ytv')]
54
+ # ('hop_dln','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
57
  # count_result = results[0].boxes.cls[0].item()