fix checkpoint (#21874)

2023-03-02 08:47:20 +01:00
parent 89359e4c63
commit 43299c63ca
1 changed files with 1 additions and 1 deletions
--- 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"])
    ```