Add support for EDS-NLP library

Author: percevalw

packages/tasks/src/library-to-tasks.ts

@@ -18,6 +18,7 @@ export const LIBRARY_TASK_MAPPING: Partial<Record<ModelLibraryKey, PipelineType[
 	bertopic: ["text-classification"],
 	diffusers: ["image-to-image", "text-to-image"],
 	doctr: ["object-detection"],
+	edsnlp: ["token-classification"],
 	espnet: ["text-to-speech", "automatic-speech-recognition"],
 	fairseq: ["text-to-speech", "audio-to-audio"],
 	fastai: ["image-classification"],

packages/tasks/src/model-libraries-snippets.ts

@@ -96,6 +96,24 @@ export const diffusers = (model: ModelData): string[] => {
 	}
 };
 
+export const edsnlp = (model: ModelData): string[] => {
+	const packageName = nameWithoutNamespace(model.id).replaceAll("-", "_");
+	return [
+		`# Load it from the huggingface hub directly
+import edsnlp
+nlp = edsnlp.load("${model.id}")
+`,
+		`# Or install it as a package
+!pip install git+https://huggingface.co/${model.id}
+
+# and import it as a module
+import ${packageName}
+
+nlp = ${packageName}.load()  # or edsnlp.load("${packageName}")
+`,
+	];
+};
+
 export const espnetTTS = (model: ModelData): string[] => [
 	`from espnet2.bin.tts_inference import Text2Speech
 

packages/tasks/src/model-libraries.ts

@@ -116,6 +116,16 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		repoName: "doctr",
 		repoUrl: "https://github.com/mindee/doctr",
 	},
+	edsnlp: {
+		prettyLabel: "EDS-NLP",
+		repoName: "edsnlp",
+		repoUrl: "https://github.com/aphp/edsnlp",
+		filter: false,
+		snippets: snippets.edsnlp,
+		countDownloads: {
+			wildcard: { path: "*/config.cfg" },
+		},
+	},
 	elm: {
 		prettyLabel: "ELM",
 		repoName: "elm",