Hugging Face's logo

Spaces:
xiaoming32236046
/
yolov10_CTC
Sleeping

App Files Community
xiaoming32236046 commited on
Commit
a8e499a
·
verified ·
1 Parent(s): 553abbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from ultralytics import YOLOv10
6
 
7
  def yolov10_inference(image, video, model_id, image_size, conf_threshold):
8
  #model = YOLOv10.from_pretrained(f'jameslahm/{model_id}')
9
- model = YOLOv10("/xiaoming32236046/yolov10_CTC/best.pt")
10
  if image:
11
  results = model.predict(source=image, imgsz=image_size, conf=conf_threshold)
12
  annotated_image = results[0].plot()
 
6
 
7
  def yolov10_inference(image, video, model_id, image_size, conf_threshold):
8
  #model = YOLOv10.from_pretrained(f'jameslahm/{model_id}')
9
+ model = YOLOv10("best.pt")
10
  if image:
11
  results = model.predict(source=image, imgsz=image_size, conf=conf_threshold)
12
  annotated_image = results[0].plot()