@huggingface/tasks 0.0.6 → 0.0.8

package/src/index.ts

@@ -1,5 +1,8 @@
-export type { TaskData, TaskDemo, TaskDemoEntry, ExampleRepo } from "./Types";
-export { TASKS_DATA } from "./tasksData";
+export { LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS } from "./library-to-tasks";
+export { MODEL_LIBRARIES_UI_ELEMENTS } from "./library-ui-elements";
+export { MAPPING_DEFAULT_WIDGET } from "./default-widget-inputs";
+export type { TaskData, TaskDemo, TaskDemoEntry, ExampleRepo } from "./tasks";
+export * from "./tasks";
 export {
 	PIPELINE_DATA,
 	PIPELINE_TYPES,
@@ -11,7 +14,35 @@ export {
 	SUBTASK_TYPES,
 	PIPELINE_TYPES_SET,
 } from "./pipelines";
-export { ModelLibrary, ALL_DISPLAY_MODEL_LIBRARY_KEYS } from "./modelLibraries";
-export type { ModelLibraryKey } from "./modelLibraries";
+export { ModelLibrary, ALL_DISPLAY_MODEL_LIBRARY_KEYS } from "./model-libraries";
+export type { ModelLibraryKey } from "./model-libraries";
+export {
+	ModelData,
+	TransformersInfo,
+	WidgetExample,
+	WidgetExampleAttribute,
+	WidgetExampleAssetAndPromptInput,
+	WidgetExampleAssetAndTextInput,
+	WidgetExampleAssetAndZeroShotInput,
+	WidgetExampleAssetInput,
+	WidgetExampleSentenceSimilarityInput,
+	WidgetExampleStructuredDataInput,
+	WidgetExampleTableDataInput,
+	WidgetExampleTextAndContextInput,
+	WidgetExampleTextAndTableInput,
+	WidgetExampleTextInput,
+	WidgetExampleZeroShotTextInput,
+	WidgetExampleOutput,
+	WidgetExampleOutputUrl,
+	WidgetExampleOutputLabels,
+	WidgetExampleOutputAnswerScore,
+	WidgetExampleOutputText,
+} from "./model-data";
+export { InferenceDisplayability } from "./model-data";
 
 export { TAG_NFAA_CONTENT, OTHER_TAGS_SUGGESTIONS, TAG_TEXT_GENERATION_INFERENCE, TAG_CUSTOM_CODE } from "./tags";
+
+import * as snippets from "./snippets";
+export { snippets };
+
+export type { LibraryUiElement } from "./library-ui-elements";

package/src/library-to-tasks.ts

@@ -0,0 +1,47 @@
+import type { ModelLibraryKey } from "./model-libraries";
+import type { PipelineType } from "./pipelines";
+
+/**
+ * Mapping from library name (excluding Transformers) to its supported tasks.
+ * Inference API should be disabled for all other (library, task) pairs beyond this mapping.
+ * As an exception, we assume Transformers supports all inference tasks.
+ * This mapping is generated automatically by "python-api-export-tasks" action in huggingface/api-inference-community repo upon merge.
+ * Ref: https://github.com/huggingface/api-inference-community/pull/158
+ */
+export const LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS: Partial<Record<ModelLibraryKey, PipelineType[]>> = {
+	"adapter-transformers": ["question-answering", "text-classification", "token-classification"],
+	allennlp: ["question-answering"],
+	asteroid: [
+		// "audio-source-separation",
+		"audio-to-audio",
+	],
+	bertopic: ["text-classification"],
+	diffusers: ["image-to-image", "text-to-image"],
+	doctr: ["object-detection"],
+	espnet: ["text-to-speech", "automatic-speech-recognition"],
+	fairseq: ["text-to-speech", "audio-to-audio"],
+	fastai: ["image-classification"],
+	fasttext: ["feature-extraction", "text-classification"],
+	flair: ["token-classification"],
+	k2: ["automatic-speech-recognition"],
+	keras: ["image-classification"],
+	nemo: ["automatic-speech-recognition"],
+	open_clip: ["zero-shot-classification", "zero-shot-image-classification"],
+	paddlenlp: ["conversational", "fill-mask", "summarization", "zero-shot-classification"],
+	peft: ["text-generation"],
+	"pyannote-audio": ["automatic-speech-recognition"],
+	"sentence-transformers": ["feature-extraction", "sentence-similarity"],
+	sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
+	spacy: ["token-classification", "text-classification", "sentence-similarity"],
+	"span-marker": ["token-classification"],
+	speechbrain: [
+		"audio-classification",
+		"audio-to-audio",
+		"automatic-speech-recognition",
+		"text-to-speech",
+		"text2text-generation",
+	],
+	stanza: ["token-classification"],
+	timm: ["image-classification"],
+	mindspore: ["image-classification"],
+};