From 2605273579b701636c1914aacfbb64c1fed67e8f Mon Sep 17 00:00:00 2001 From: Luke Southam Date: Sun, 11 Dec 2022 02:01:52 +0000 Subject: [PATCH] update --- whispercpp.pyx | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/whispercpp.pyx b/whispercpp.pyx index edb5849..3ae3d2e 100644 --- a/whispercpp.pyx +++ b/whispercpp.pyx @@ -16,11 +16,11 @@ cdef char* DEFAULT_MODEL = b'tiny' cdef char* LANGUAGE = b'fr' MODELS = { - b'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', - b'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', - b'model_ggml_large.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-large.bin', + 'model_ggml_tiny.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-tiny.bin', + 'model_ggml_base.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-base.bin', + 'model_ggml_small.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-small.bin', + 'model_ggml_medium.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-medium.bin', + 'model_ggml_large.bin': 'https://ggml.ggerganov.com/ggml-model-whisper-large.bin', } def model_exists(model): @@ -31,7 +31,7 @@ def download_model(model): return print(f'Downloading {model}...') - url = MODELS[model] + url = MODELS[model.decode()] r = requests.get(url, allow_redirects=True) with open(model, 'wb') as f: f.write(r.content)