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];