diff --git a/ultralytics/models/yolov10/card.py b/ultralytics/models/yolov10/card.py index e87820ad..fc2405c4 100644 --- a/ultralytics/models/yolov10/card.py +++ b/ultralytics/models/yolov10/card.py @@ -1,6 +1,8 @@ card_template_text = """ --- license: agpl-3.0 +library_name: ultralytics +repo_url: https://github.com/THU-MIG/yolov10 tags: - object-detection - computer-vision diff --git a/ultralytics/models/yolov10/model.py b/ultralytics/models/yolov10/model.py index 123e31b0..09592c83 100644 --- a/ultralytics/models/yolov10/model.py +++ b/ultralytics/models/yolov10/model.py @@ -7,9 +7,7 @@ from .train import YOLOv10DetectionTrainer from huggingface_hub import PyTorchModelHubMixin from .card import card_template_text -class YOLOv10(Model, PyTorchModelHubMixin, library_name="ultralytics", \ - repo_url="https://github.com/THU-MIG/yolov10", \ - model_card_template=card_template_text): +class YOLOv10(Model, PyTorchModelHubMixin, model_card_template=card_template_text): def __init__(self, model="yolov10n.pt", task=None, verbose=False, names=None):