feat: workflow

This commit is contained in:
2024-08-01 14:45:54 +08:00
parent 0312646819
commit 30dedc935f
4 changed files with 83 additions and 21 deletions

View File

@ -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