hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

compatible with local whisper model

Browse Source
This commit is contained in:
harry
2024-04-11 18:27:14 +08:00
parent 759c1ce652
commit 1c7d7e138b
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/services/subtitle.py
View File

@@ -1,4 +1,5 @@
import json import json
import os.path
import re import re
from faster_whisper import WhisperModel from faster_whisper import WhisperModel
@@ -17,8 +18,13 @@ model = None
def create(audio_file, subtitle_file: str = ""): def create(audio_file, subtitle_file: str = ""):
global model global model
if not model: if not model:
logger.info(f"loading model: {model_size}, device: {device}, compute_type: {compute_type}") model_path = f"{utils.root_dir()}/models/whisper-{model_size}"
model = WhisperModel(model_size_or_path=model_size, model_bin_file = f"{model_path}/model.bin"
if not os.path.isdir(model_path) or not os.path.isfile(model_bin_file):
model_path = model_size
logger.info(f"loading model: {model_path}, device: {device}, compute_type: {compute_type}")
model = WhisperModel(model_size_or_path=model_path,
device=device, device=device,
compute_type=compute_type) compute_type=compute_type)