From 43299c63cab44e44df4c42d6cc93fd4b6729d866 Mon Sep 17 00:00:00 2001 From: Arthur <48595927+ArthurZucker@users.noreply.github.com> Date: Thu, 2 Mar 2023 08:47:20 +0100 Subject: [PATCH] fix checkpoint (#21874) --- src/transformers/pipelines/zero_shot_audio_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/zero_shot_audio_classification.py b/src/transformers/pipelines/zero_shot_audio_classification.py index 497e7968eb..c534763e87 100644 --- a/src/transformers/pipelines/zero_shot_audio_classification.py +++ b/src/transformers/pipelines/zero_shot_audio_classification.py @@ -32,7 +32,7 @@ class ZeroShotAudioClassificationPipeline(ChunkPipeline): >>> dataset = load_dataset("ashraq/esc50") >>> audio = next(iter(dataset["train"]["audio"]))["array"] - >>> classifier = pipeline(task="zero-shot-audio-classification", model="laion-ai/clap-htsat-tiny") + >>> classifier = pipeline(task="zero-shot-audio-classification", model="laion/clap-htsat-unfused") >>> classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"]) ```