xiaoming32236046 commited on
Commit
7bc3a5f
·
verified ·
1 Parent(s): 21415d2

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("/ddn/imu_tch1/project/yolov10/yolov10-1.0/runs/detect/train9/weights/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()