Skip to content

Commit

Permalink
update huggingface hub card template
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslahm committed Jun 7, 2024
1 parent e161d70 commit ea93d4f
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 2 additions & 0 deletions ultralytics/models/yolov10/card.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
4 changes: 1 addition & 3 deletions ultralytics/models/yolov10/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,7 @@
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):
Expand Down

0 comments on commit ea93d4f

Please sign in to comment.