File size: 437 Bytes
9de729b 8a965da 9de729b 973bb39 33d11d7 7e8d389 d9a32db db75012 0a7507f 68ed0e8 9de729b |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# TEST MODEL
from transformers import pipeline
classifier = pipeline(task="zero-shot-audio-classification", model="mskov/roberta-base-toxicity")
# classifier = pipeline(model="mskov/roberta-base-toxicity")
audio = "./test/candy-bar-chewing.wav"
classifier(audio, candidate_labels=["Sound of a dog", "Sound of vaccum cleaner", "chewing", "sneezing"])
predicted_label = result[0]["label"]
print(f"Predicted label: {predicted_label}") |