Skip to content

Commit 84d698a

Browse files
authored
fix: yolo fix cli.py with project id (#711)
1 parent 2c69bb3 commit 84d698a

File tree

1 file changed

+1
-1
lines changed
  • label_studio_ml/examples/yolo

1 file changed

+1
-1
lines changed

label_studio_ml/examples/yolo/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def run(self, project, tasks):
5656

5757
# load YOLO model
5858
# TODO: use get_all_classes_inherited_LabelStudioMLBase to detect model classes
59-
model = YOLO(project_id=project.id, label_config=project.label_config)
59+
model = YOLO(project_id=str(project.id), label_config=project.label_config)
6060
logger.info(f"YOLO ML backend is created")
6161

6262
# predict and send prediction to Label Studio

0 commit comments

Comments
 (0)