diff --git a/tests/pipelines/test_pipelines_zero_shot_audio_classification.py b/tests/pipelines/test_pipelines_zero_shot_audio_classification.py index 2790f66a15..80414f1722 100644 --- a/tests/pipelines/test_pipelines_zero_shot_audio_classification.py +++ b/tests/pipelines/test_pipelines_zero_shot_audio_classification.py @@ -52,7 +52,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase): model="laion/clap-htsat-unfused", ) # This is an audio of a dog - dataset = load_dataset("ashraq/esc50") + dataset = load_dataset("hf-internal-testing/ashraq-esc50-1-dog-example") audio = dataset["train"]["audio"][-1]["array"] output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])