diff --git a/packages/tasks/src/pipelines.ts b/packages/tasks/src/pipelines.ts index 379842b04..c3f051986 100644 --- a/packages/tasks/src/pipelines.ts +++ b/packages/tasks/src/pipelines.ts @@ -70,6 +70,12 @@ export interface PipelineData { /// This will impact the default pipeline tag of a model when not /// specified. export const PIPELINE_DATA = { + "biology": { + name: "Biology", + modality: "multimodal", + color: "green", + hideInDatasets: true, + }, "text-classification": { name: "Text Classification", subtasks: [ diff --git a/packages/tasks/src/tasks/index.ts b/packages/tasks/src/tasks/index.ts index b0c78ebdd..79c954d8e 100644 --- a/packages/tasks/src/tasks/index.ts +++ b/packages/tasks/src/tasks/index.ts @@ -117,6 +117,7 @@ export const TASKS_MODEL_LIBRARIES: Record = { "audio-to-audio": ["asteroid", "fairseq", "speechbrain"], "automatic-speech-recognition": ["espnet", "nemo", "speechbrain", "transformers", "transformers.js"], "audio-text-to-text": [], + "biology": [], "depth-estimation": ["transformers", "transformers.js"], "document-question-answering": ["transformers", "transformers.js"], "feature-extraction": ["sentence-transformers", "transformers", "transformers.js"], @@ -200,6 +201,7 @@ export const TASKS_DATA: Record = { "audio-classification": getData("audio-classification", audioClassification), "audio-to-audio": getData("audio-to-audio", audioToAudio), "audio-text-to-text": getData("audio-text-to-text", placeholder), + "biology": getData("biology", placeholder), "automatic-speech-recognition": getData("automatic-speech-recognition", automaticSpeechRecognition), "depth-estimation": getData("depth-estimation", depthEstimation), "document-question-answering": getData("document-question-answering", documentQuestionAnswering),