This commit is contained in:
yumoqing 2024-08-03 17:57:04 +08:00
parent 642e2f1ab7
commit f8dc17f12f
5 changed files with 26 additions and 33 deletions

View File

@ -4,7 +4,8 @@ from appPublic.log import MyLogger, info, debug, warning
from appPublic.folderUtils import ProgramPath from appPublic.folderUtils import ProgramPath
from appPublic.jsonConfig import getConfig from appPublic.jsonConfig import getConfig
from ahserver.configuredServer import ConfiguredServer from ahserver.configuredServer import ConfiguredServer
import asr_engine from ahserver.serverenv import ServerEnv
from whisper_model import WhisperBase64
__version__ = '0.0.1' __version__ = '0.0.1'
@ -13,6 +14,7 @@ if __name__ == '__main__':
parser.add_argument('-w', '--workdir') parser.add_argument('-w', '--workdir')
parser.add_argument('-p', '--port') parser.add_argument('-p', '--port')
args = parser.parse_args() args = parser.parse_args()
print(args)
workdir = args.workdir or os.getcwd() workdir = args.workdir or os.getcwd()
p = ProgramPath() p = ProgramPath()
config = getConfig(workdir, NS={'workdir':workdir, 'ProgramPath':p}) config = getConfig(workdir, NS={'workdir':workdir, 'ProgramPath':p})

View File

@ -1,17 +1,12 @@
from ahserver.serverenv import ServerEnv from ahserver.serverenv import ServerEnv
from aiohttp.web import StreamResponse from aiohttp.web import StreamResponse
from whisper.load_model import WhisperBase64
async def generate(request, **kw): async def generate(request, **kw):
model = kw.get('model', 'whisper') model = kw.get('model', 'whisper')
engine = None engine = None
g = ServerEnv() g = ServerEnv()
if model=='whisper': if model=='whisper':
engine = g.whisper_engine engine = g.whisper_engine
if engine is None:
engine = WhisperBase64
g.whisper_engine = engine
await resp.prepare(request) await resp.prepare(request)
line = await request.content.readline() line = await request.content.readline()

View File

@ -1,2 +0,0 @@
openai-whisper

View File

@ -1,6 +1,7 @@
from ahserver.serverenv import ServerEnv from ahserver.serverenv import ServerEnv
from appPublic.worker import awaitify from appPublic.worker import awaitify
from appPublic.jsonConfig import getConfig, get_definition
from ahserver.globalEnv import get_definition
import numpy as np import numpy as np
import base64 import base64
import whisper import whisper
@ -40,6 +41,3 @@ class WhisperBase64(WhisperBase):
ndarr = np.frombuffer(raw, dtype=np.float32) ndarr = np.frombuffer(raw, dtype=np.float32)
return self.model.transcribe(raw) return self.model.transcribe(raw)
g = ServerEnv()
g.whisper_engine = WhisperBase64()