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 <lucainp@gmail.com> Co-authored-by: Lucain <lucain@huggingface.co>
huggingface · Jan 15, 2025 · 1a20326 · 1a20326
1 parent c999114
commit 1a20326
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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",