Sync widgets demo

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

@@ -31,6 +31,7 @@ export const LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS: Partial<Record<ModelLi
 	peft: ["text-generation"],
 	"pyannote-audio": ["automatic-speech-recognition"],
 	"sentence-transformers": ["feature-extraction", "sentence-similarity"],
+	setfit: ["text-classification"],
 	sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
 	spacy: ["token-classification", "text-classification", "sentence-similarity"],
 	"span-marker": ["token-classification"],

packages/tasks/src/library-ui-elements.ts

@@ -358,6 +358,12 @@ const sentenceTransformers = (model: ModelData) => [
 model = SentenceTransformer("${model.id}")`,
 ];
 
+const setfit = (model: ModelData) => [
+	`from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`,
+];
+
 const spacy = (model: ModelData) => [
 	`!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
 
@@ -661,6 +667,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS: Partial<Record<ModelLibraryKey, Librar
 		docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
 		snippets: sentenceTransformers,
 	},
+	setfit: {
+		btnLabel: "setfit",
+		repoName: "setfit",
+		repoUrl: "https://github.com/huggingface/setfit",
+		docsUrl: "https://huggingface.co/docs/hub/setfit",
+		snippets: setfit,
+	},
 	sklearn: {
 		btnLabel: "Scikit-learn",
 		repoName: "Scikit-learn",

packages/tasks/src/model-libraries.ts

@@ -23,6 +23,7 @@ export enum ModelLibrary {
 	"pyannote-audio" = "pyannote.audio",
 	"sample-factory" = "Sample Factory",
 	"sentence-transformers" = "Sentence Transformers",
+	"setfit" = "SetFit",
 	"sklearn" = "Scikit-learn",
 	"spacy" = "spaCy",
 	"span-marker" = "SpanMarker",

packages/tasks/src/tasks/index.ts

@@ -69,7 +69,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
 	"tabular-classification": ["sklearn"],
 	"tabular-regression": ["sklearn"],
 	"tabular-to-text": ["transformers"],
-	"text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+	"text-classification": ["adapter-transformers", "setfit", "spacy", "transformers", "transformers.js"],
 	"text-generation": ["transformers", "transformers.js"],
 	"text-retrieval": [],
 	"text-to-image": ["diffusers"],