Downloads · 30 days
14
2% of all-time downloads
shmart/shmisper-medium-PL
shmisper-medium-PL is a machine learning model from shmart. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
This model is a result of finetuning openai/whisper-medium model on custom PL dataset and then conversion to faster-whisper model. In training dataset there were also 5 english speakers and 4 japanese speakers for whi…
Downloads · 30 days
14
2% of all-time downloads
All-time downloads
739
Public
Repo size
4.6 GB
Likes
0
Public
Click a slice to open those files.
.bin1.5 GB · 100%
From the Hugging Face model README
This model is a result of finetuning openai/whisper-medium model on custom PL dataset and then conversion to faster-whisper model.
In training dataset there were also 5 english speakers and 4 japanese speakers for which polish transcription was manually created.
from faster_whisper import WhisperModel
import huggingface_hub
model_path = huggingface_hub.snapshot_download("shmart/shmisper-medium-PL")
model = WhisperModel(model_path, device="cuda", compute_type="float16")
options = {
'language': "pl",
'beam_size': 5,
'without_timestamps': True,
'suppress_tokens': [],
'log_prob_threshold': None,
'no_speech_threshold': 0.05
}
input_wav_path = './audio.wav'
result, info = model.transcribe(input_wav_path, **options)
text = ' '.join([r.text for r in result])
print(text)