|
|
--- |
|
|
license: cc |
|
|
--- |
|
|
# Whisper-base-minds14-ForClassification |
|
|
Dataset source: https://huggingface.co/datasets/PolyAI/minds14 <br> |
|
|
Model used for Fine Tuning: https://huggingface.co/openai/whisper-base <br> |
|
|
Trained on Colab T4 @ 5 epoch <br> |
|
|
|
|
|
Usage: |
|
|
```python |
|
|
from transformers import WhisperProcessor, WhisperForAudioClassification |
|
|
processor = WhisperProcessor.from_pretrained("Willy030125/Whisper-base-minds14-ForClassification") |
|
|
model = WhisperForAudioClassification.from_pretrained("Willy030125/Whisper-base-minds14-ForClassification", num_labels=14) |
|
|
``` |