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

use a more standard model storage path, and make the directory if doesn't exist

Browse Source
This commit is contained in:
Rasmus Larsen 2023-01-30 17:30:18 +01:00
parent 5157b341a0
commit 4526cce0ff
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
whispercpp.pyx
View File

@ -7,7 +7,7 @@ import requests
import os import os
from pathlib import Path from pathlib import Path
MODELS_DIR = str(Path('~/ggml-models').expanduser()) MODELS_DIR = str(Path('~/.ggml-models').expanduser())
print("Saving models to:", MODELS_DIR) print("Saving models to:", MODELS_DIR)
@ -37,6 +37,7 @@ def download_model(model):
print(f'Downloading {model}...') print(f'Downloading {model}...')
url = MODELS[model.decode()] url = MODELS[model.decode()]
r = requests.get(url, allow_redirects=True) r = requests.get(url, allow_redirects=True)
os.makedirs(MODELS_DIR, exist_ok=True)
with open(MODELS_DIR + "/" + model.decode(), 'wb') as f: with open(MODELS_DIR + "/" + model.decode(), 'wb') as f:
f.write(r.content) f.write(r.content)
@ -78,7 +79,6 @@ cdef whisper_full_params default_params() nogil:
params.print_realtime = True params.print_realtime = True
params.print_progress = True params.print_progress = True
params.translate = False params.translate = False
params.language = <const char *> LANGUAGE
n_threads = N_THREADS n_threads = N_THREADS
return params return params