ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-14 00:53:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

add blackbox fastchat

Browse Source
This commit is contained in:
ACBBZ
2024-04-10 08:56:18 +00:00
parent ec10081276
commit e2106fc33c
2 changed files with 78 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/blackbox/blackbox_factory.py
View File

@ -7,8 +7,9 @@ from .blackbox import Blackbox
from .calculator import Calculator
from .text_to_audio import TextToAudio
from .tesou import Tesou
from .fastcaht import Fastchat
class BlackboxFactory:
class BlackboxFactor:
def __init__(self) -> None:
self.tts = TTS()
@ -19,6 +20,7 @@ class BlackboxFactory:
self.audio_to_text = AudioToText()
self.text_to_audio = TextToAudio()
self.tesou = Tesou()
self.fastchat = Fastchat()
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)
@ -40,4 +42,6 @@ class BlackboxFactory:
return self.sum
if blackbox_name == "tesou":
return self.tesou
if blackbox_name == "fastchat":
return self.fastchat
raise ValueError("Invalid blockbox type")