bugfix
This commit is contained in:
parent
642e2f1ab7
commit
f8dc17f12f
@ -4,7 +4,8 @@ from appPublic.log import MyLogger, info, debug, warning
|
||||
from appPublic.folderUtils import ProgramPath
|
||||
from appPublic.jsonConfig import getConfig
|
||||
from ahserver.configuredServer import ConfiguredServer
|
||||
import asr_engine
|
||||
from ahserver.serverenv import ServerEnv
|
||||
from whisper_model import WhisperBase64
|
||||
|
||||
__version__ = '0.0.1'
|
||||
|
||||
@ -13,6 +14,7 @@ if __name__ == '__main__':
|
||||
parser.add_argument('-w', '--workdir')
|
||||
parser.add_argument('-p', '--port')
|
||||
args = parser.parse_args()
|
||||
print(args)
|
||||
workdir = args.workdir or os.getcwd()
|
||||
p = ProgramPath()
|
||||
config = getConfig(workdir, NS={'workdir':workdir, 'ProgramPath':p})
|
||||
|
@ -1,17 +1,12 @@
|
||||
from ahserver.serverenv import ServerEnv
|
||||
from aiohttp.web import StreamResponse
|
||||
|
||||
from whisper.load_model import WhisperBase64
|
||||
|
||||
async def generate(request, **kw):
|
||||
model = kw.get('model', 'whisper')
|
||||
engine = None
|
||||
g = ServerEnv()
|
||||
if model=='whisper':
|
||||
engine = g.whisper_engine
|
||||
if engine is None:
|
||||
engine = WhisperBase64
|
||||
g.whisper_engine = engine
|
||||
|
||||
await resp.prepare(request)
|
||||
line = await request.content.readline()
|
||||
|
@ -1,2 +0,0 @@
|
||||
openai-whisper
|
||||
|
@ -1,6 +1,7 @@
|
||||
from ahserver.serverenv import ServerEnv
|
||||
from appPublic.worker import awaitify
|
||||
from appPublic.jsonConfig import getConfig, get_definition
|
||||
|
||||
from ahserver.globalEnv import get_definition
|
||||
import numpy as np
|
||||
import base64
|
||||
import whisper
|
||||
@ -40,6 +41,3 @@ class WhisperBase64(WhisperBase):
|
||||
ndarr = np.frombuffer(raw, dtype=np.float32)
|
||||
return self.model.transcribe(raw)
|
||||
|
||||
g = ServerEnv()
|
||||
g.whisper_engine = WhisperBase64()
|
||||
|
Loading…
Reference in New Issue
Block a user