Add MITIE as library

huggingface · Jan 15, 2025 · 0f1d251 · 0f1d251
1 parent 529248a
commit 0f1d251
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_extension:dat) AND (path_filename:mitie_model* OR path_filename:total_word_feature_extractor)",
+	},
 	"ml-agents": {
 		prettyLabel: "ml-agents",
 		repoName: "ml-agents",