From 1a20326df5308b9b359bfa8782239d6cc18897f8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tymon=20=C5=BBarski?= <68420753+tymzar@users.noreply.github.com> Date: Wed, 15 Jan 2025 16:43:10 +0100 Subject: [PATCH] Add MITIE as library (#1105) This PR ensures download stats work for the [MITIE](https://github.com/mit-nlp/MITIE) models. I would much appreciate it if you would consider the following change! --------- Co-authored-by: Lucain Co-authored-by: Lucain --- packages/tasks/src/model-libraries.ts | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 3faa581c1..4e7a20c51 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -471,6 +471,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { countDownloads: `path:"MeshAnything_350m.pth"`, snippets: snippets.mesh_anything, }, + mitie: { + prettyLabel: "MITIE", + repoName: "MITIE", + repoUrl: "https://github.com/mit-nlp/MITIE", + countDownloads: `path_filename:"total_word_feature_extractor"`, + }, "ml-agents": { prettyLabel: "ml-agents", repoName: "ml-agents",