Hugging Face's logo

Spaces:
trannam1084
/
Object_Detection
Sleeping

App Files Community
trannam1084 commited on
Commit
064f359
·
verified ·
1 Parent(s): d26187e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  import supervision as sv
7
  from ultralytics import YOLO
8
 
9
- model = YOLO("yolov8n.pt")
10
  CLASS_NAMES_DICT = model.model.names
11
 
12
  SELECTED_CLASS_NAMES = ['car', 'bus', 'truck', 'motorcycle']
 
6
  import supervision as sv
7
  from ultralytics import YOLO
8
 
9
+ model = YOLO("yolo11n.pt")
10
  CLASS_NAMES_DICT = model.model.names
11
 
12
  SELECTED_CLASS_NAMES = ['car', 'bus', 'truck', 'motorcycle']