Spaces:

xiaoming32236046
/

yolov10_CTC

Running

xiaoming32236046 commited on 18 days ago

Commit

a8e499a

•

1 Parent(s): 553abbc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from ultralytics import YOLOv10
 def yolov10_inference(image, video, model_id, image_size, conf_threshold):
     #model = YOLOv10.from_pretrained(f'jameslahm/{model_id}')
-    model = YOLOv10("/xiaoming32236046/yolov10_CTC/best.pt")
     if image:
         results = model.predict(source=image, imgsz=image_size, conf=conf_threshold)
         annotated_image = results[0].plot()

 def yolov10_inference(image, video, model_id, image_size, conf_threshold):
     #model = YOLOv10.from_pretrained(f'jameslahm/{model_id}')
+    model = YOLOv10("best.pt")
     if image:
         results = model.predict(source=image, imgsz=image_size, conf=conf_threshold)
         annotated_image = results[0].plot()