From 234a87f018c7f3637e2b4534ccf57e820679fc0d Mon Sep 17 00:00:00 2001 From: Vaibhavs10 Date: Fri, 20 Jun 2025 13:59:21 +0200 Subject: [PATCH] Add Zero-shot-image-classification and image-classification. --- packages/tasks/src/model-libraries-snippets.ts | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 212686160e..f777d18ee9 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -1280,6 +1280,15 @@ export const transformers = (model: ModelData): string[] => { pipelineSnippet.push("messages = [", ' {"role": "user", "content": "Who are you?"},', "]"); pipelineSnippet.push("pipe(messages)"); } + } else if (model.pipeline_tag === "zero-shot-image-classification") { + pipelineSnippet.push( + "pipe(", + ' "https://huggingface.co/datasets/Narsil/image_dummy/raw/main/parrots.png",', + ' candidate_labels=["animals", "humans", "landscape"],', + ")" + ); + } else if (model.pipeline_tag === "image-classification") { + pipelineSnippet.push('pipe("https://huggingface.co/datasets/Narsil/image_dummy/raw/main/parrots.png")'); } return [pipelineSnippet.join("\n"), autoSnippet];