vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-05-01 21:20:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Alpine fix version, STT fixes

Browse source
This commit is contained in:
frdel 2024-11-29 08:55:27 +01:00
parent 81e653ba2d
commit c99b1a47d4
5 changed files with 52 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

10
python/helpers/whisper.py
View file

@ -3,16 +3,20 @@ import base64
import warnings
import whisper
import tempfile
from python.helpers import runtime, rfc
from python.helpers import runtime, rfc, settings
# suppress FutureWarning from torch.load
warnings.filterwarnings('ignore', category=FutureWarning)
model = None
model_name = ""
def preload():
global model
model = whisper.load_model("base")
global model, model_name
set = settings.get_settings()
if not model or model_name != set["stt_model_size"]:
model = whisper.load_model(set["stt_model_size"])
model_name = set["stt_model_size"]
return model
async def transcribe(audio_bytes_b64: str):