Downloads · 30 days
10
4% of all-time downloads
xb2g/thai-whisper-small
thai-whisper-small is a machine learning model from xb2g. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Fine-tuned Whisper Small model for Thai Automatic Speech Recognition.
Downloads · 30 days
10
4% of all-time downloads
All-time downloads
233
Public
Parameters
242M
1.9 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt967 MB · 50%
From the Hugging Face model README
Fine-tuned Whisper Small model for Thai Automatic Speech Recognition.
| Epoch | Train Loss | Test Loss | Improvement |
|---|---|---|---|
| 1 | 1.9345 | 0.2274 | Baseline |
| 2 | 0.2893 | 0.2328 | -85% |
| 3 | 0.2691 | 0.2217 | -86% |
import torch
from transformers import WhisperProcessor, WhisperForConditionalGeneration
# Load processor from base model
processor = WhisperProcessor.from_pretrained("openai/whisper-small")
# Load base model
model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
# Load fine-tuned weights
model.load_state_dict(torch.load("model.pt", map_location="cpu"))
model.eval()
MIT License