From e32f38df4d8b0a6d8a639384a33d5ce5d9127984 Mon Sep 17 00:00:00 2001
From: Merve Noyan <mervenoyan@Merve-MacBook-Pro.local>
Date: Thu, 9 Jan 2025 10:42:47 +0100
Subject: [PATCH] add path ext to yolov8 and yolov10

---
 packages/tasks/src/model-libraries.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts
index b8bb95ccc..e4145cede 100644
--- a/packages/tasks/src/model-libraries.ts
+++ b/packages/tasks/src/model-libraries.ts
@@ -849,6 +849,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		repoUrl: "https://github.com/ultralytics/ultralytics",
 		docsUrl: "https://github.com/ultralytics/ultralytics",
 		filter: false,
+		countDownloads: `path_extension:"pt"`,
 		snippets: snippets.yolov8,
 	},
 	yolov10: {
@@ -856,6 +857,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		repoName: "yolov10",
 		repoUrl: "https://github.com/THU-MIG/yolov10",
 		docsUrl: "https://github.com/THU-MIG/yolov10",
+		countDownloads: `path_extension:"pt"`,
 		snippets: snippets.yolov10,
 	},
 	whisperkit: {