bmombie
/

bicycle_rider_detector_002

Object Detection

Inference Endpoints

Model card Files Files and versions Community

bmombie commited on Apr 23

Commit

e916ed1

•

1 Parent(s): 9a2b60b

Update main.py

Files changed (1) hide show

main.py +1 -1

main.py CHANGED Viewed

@@ -10,7 +10,7 @@ from ultralytics import YOLO
 # Initialize your model
 def init_context(context):
 	context.logger.info('Init context...  0%')
-	model = YOLO('custom-yolov8n.pt')
 	context.user_data.model_handler = model
 	context.logger.info('Init context...100%')

 # Initialize your model
 def init_context(context):
 	context.logger.info('Init context...  0%')
+	model = YOLO('best.pt')
 	context.user_data.model_handler = model
 	context.logger.info('Init context...100%')