mirror of
https://github.com/BoardWare-Genius/jarvis-models.git
synced 2025-12-13 16:53:24 +00:00
wip
This commit is contained in:
2
main.py
2
main.py
@ -20,7 +20,7 @@ class Main():
|
||||
def run(self):
|
||||
logger = logging.getLogger(__name__)
|
||||
logger.info("jarvis-models start", extra={"version": "0.0.1"})
|
||||
uvicorn.run("server:app", host="0.0.0.0", port=8000, log_level="info")
|
||||
uvicorn.run("server:app", host="0.0.0.0", port=8001, log_level="info")
|
||||
|
||||
if __name__ == "__main__":
|
||||
injector = Injector()
|
||||
|
||||
@ -4,6 +4,8 @@ from typing import Any, Coroutine
|
||||
from fastapi import Request, Response, status
|
||||
from fastapi.responses import JSONResponse
|
||||
|
||||
from ..log.log import Logger
|
||||
|
||||
from ..asr.rapid_paraformer.utils import read_yaml
|
||||
from ..asr.rapid_paraformer import RapidParaformer
|
||||
from .blackbox import Blackbox
|
||||
@ -13,7 +15,8 @@ from injector import singleton, inject
|
||||
class ASR(Blackbox):
|
||||
|
||||
@inject
|
||||
def __init__(self,path = ".env.yaml") -> None:
|
||||
def __init__(self,logger: Logger,path = ".env.yaml") -> None:
|
||||
logger.info("ASR init")
|
||||
config = read_yaml(path)
|
||||
self.paraformer = RapidParaformer(config)
|
||||
|
||||
|
||||
15
src/log/log.py
Normal file
15
src/log/log.py
Normal file
@ -0,0 +1,15 @@
|
||||
import logging
|
||||
from injector import singleton, inject
|
||||
|
||||
from ..configuration import LogConf
|
||||
|
||||
@singleton
|
||||
class Logger(logging.Logger):
|
||||
|
||||
@inject
|
||||
def __init__(self, config: LogConf) -> None:
|
||||
super().__init__(__name__)
|
||||
self.setLevel(config.level)
|
||||
self.addHandler(logging.StreamHandler())
|
||||
self.addHandler(logging.FileHandler('log.log'))
|
||||
|
||||
Reference in New Issue
Block a user