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

Fix support for Windows systems

Browse Source
This commit is contained in:
harry
2024-03-14 13:49:02 +08:00
parent f55ea42a07
commit 9650def321
6 changed files with 26 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/services/subtitle.py
View File

@@ -6,14 +6,14 @@ from timeit import default_timer as timer
from loguru import logger
from app.config import config
from app.models import const
from app.utils import utils
model_size = config.whisper.get("model_size", "large-v3")
device = config.whisper.get("device", "cpu")
compute_type = config.whisper.get("compute_type", "int8")
model = WhisperModel(model_size_or_path=model_size, device=device, compute_type=compute_type)
if config.app.get("subtitle_provider") == "whisper":
model = WhisperModel(model_size_or_path=model_size, device=device, compute_type=compute_type)
def create(audio_file, subtitle_file: str = ""):