lightmare/whispercpp.py
1
1
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Luke Southam 2022-12-11 02:01:52 +00:00
parent 564afa6e44
commit 2605273579
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
whispercpp.pyx
View File

@ -16,11 +16,11 @@ cdef char* DEFAULT_MODEL = b'tiny'
cdef char* LANGUAGE = b'fr' cdef char* LANGUAGE = b'fr'
MODELS = { MODELS = {
b'model_ggml_tiny.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-tiny.bin', 'model_ggml_tiny.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-tiny.bin',
b'model_ggml_base.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-base.bin', 'model_ggml_base.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-base.bin',
b'model_ggml_small.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-small.bin', 'model_ggml_small.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-small.bin',
b'model_ggml_medium.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-medium.bin', 'model_ggml_medium.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-medium.bin',
b'model_ggml_large.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-large.bin', 'model_ggml_large.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-large.bin',
} }
def model_exists(model): def model_exists(model):
@ -31,7 +31,7 @@ def download_model(model):
return return
print(f'Downloading {model}...') print(f'Downloading {model}...')
url = MODELS[model] url = MODELS[model.decode()]
r = requests.get(url, allow_redirects=True) r = requests.get(url, allow_redirects=True)
with open(model, 'wb') as f: with open(model, 'wb') as f:
f.write(r.content) f.write(r.content)