dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

This commit is contained in:
ydshieh 2024-04-17 06:49:53 +02:00
parent 0fc7d0a935
commit b01a38ea92
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pipelines/test_pipelines_zero_shot_audio_classification.py
View File

@ -52,7 +52,7 @@ class ZeroShotAudioClassificationPipelineTests(unittest.TestCase):
model="laion/clap-htsat-unfused", model="laion/clap-htsat-unfused",
) )
# This is an audio of a dog # 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"] audio = dataset["train"]["audio"][-1]["array"]
output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"]) output = audio_classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner"])