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

Update whispercpp.pyx

Browse Source
This commit is contained in:
Luke Southam 2022-12-14 17:12:39 +00:00 committed by GitHub
parent a021ae79da
commit 1e49c6627d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
whispercpp.pyx
View File

@ -7,6 +7,9 @@ import ffmpeg
import numpy as np
import requests
import os
from pathlib import Path
MODELS_DIR = Path('~/ggml-models').expanduser()
cimport numpy as cnp
@ -35,7 +38,7 @@ def download_model(model):
print(f'Downloading {model}...')
url = MODELS[model.decode()]
r = requests.get(url, allow_redirects=True)
with open(model, 'wb') as f:
with open(MODELS_DIR / model, 'wb') as f:
f.write(r.content)
@ -88,7 +91,7 @@ cdef class Whisper:
def __init__(self, model=DEFAULT_MODEL, pb=None):
model_fullname = f'model_ggml_{model}.bin'.encode('utf8')
download_model(model_fullname)
cdef bytes model_b = model_fullname
cdef bytes model_b = MODELS_DIR / model_fullname
self.ctx = whisper_init(model_b)
self.params = default_params()
whisper_print_system_info()