mirror of
https://github.com/BoardWare-Genius/jarvis-models.git
synced 2025-12-14 17:13:25 +00:00
feat: workflow
This commit is contained in:
@ -112,12 +112,24 @@ def sum_loader():
|
||||
from .sum import Sum
|
||||
return Injector().get(Sum)
|
||||
|
||||
@model_loader(lazy=blackboxConf.lazyloading)
|
||||
def audio_to_text_loader():
|
||||
from .audio_to_text import AudioToText
|
||||
return Injector().get(AudioToText)
|
||||
|
||||
@model_loader(lazy=blackboxConf.lazyloading)
|
||||
def text_to_audio_loader():
|
||||
from .text_to_audio import TextToAudio
|
||||
return Injector().get(TextToAudio)
|
||||
|
||||
@singleton
|
||||
class BlackboxFactory:
|
||||
models = {}
|
||||
|
||||
@inject
|
||||
def __init__(self,) -> None:
|
||||
self.models["text_to_audio"] = text_to_audio_loader
|
||||
self.models["audio_to_text"] = audio_to_text_loader
|
||||
self.models["asr"] = asr_loader
|
||||
self.models["tts"] = tts_loader
|
||||
self.models["sentiment_engine"] = sentiment_loader
|
||||
|
||||
Reference in New Issue
Block a user