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):
|
def run(self):
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
logger.info("jarvis-models start", extra={"version": "0.0.1"})
|
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__":
|
if __name__ == "__main__":
|
||||||
injector = Injector()
|
injector = Injector()
|
||||||
|
|||||||
@ -4,6 +4,8 @@ from typing import Any, Coroutine
|
|||||||
from fastapi import Request, Response, status
|
from fastapi import Request, Response, status
|
||||||
from fastapi.responses import JSONResponse
|
from fastapi.responses import JSONResponse
|
||||||
|
|
||||||
|
from ..log.log import Logger
|
||||||
|
|
||||||
from ..asr.rapid_paraformer.utils import read_yaml
|
from ..asr.rapid_paraformer.utils import read_yaml
|
||||||
from ..asr.rapid_paraformer import RapidParaformer
|
from ..asr.rapid_paraformer import RapidParaformer
|
||||||
from .blackbox import Blackbox
|
from .blackbox import Blackbox
|
||||||
@ -13,7 +15,8 @@ from injector import singleton, inject
|
|||||||
class ASR(Blackbox):
|
class ASR(Blackbox):
|
||||||
|
|
||||||
@inject
|
@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)
|
config = read_yaml(path)
|
||||||
self.paraformer = RapidParaformer(config)
|
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