From ea93d4f3791f7271ecdabbe381fe4187eb397833 Mon Sep 17 00:00:00 2001
From: wa22 <wa22@mails.tsinghua.edu.cn>
Date: Fri, 7 Jun 2024 16:22:42 +0800
Subject: [PATCH] update huggingface hub card template

---
 ultralytics/models/yolov10/card.py  | 2 ++
 ultralytics/models/yolov10/model.py | 4 +---
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/ultralytics/models/yolov10/card.py b/ultralytics/models/yolov10/card.py
index e87820ad5..fc2405c4a 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 123e31b05..09592c830 100644
--- a/ultralytics/models/yolov10/model.py
+++ b/ultralytics/models/yolov10/model.py
@@ -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):