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

update fastchat.py

Browse Source
This commit is contained in:
ACBBZ
2024-04-17 06:37:41 +00:00
parent 8a27cc3dea
commit 960876c21b
3 changed files with 139 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/blackbox/blackbox_factory.py
View File

@ -9,6 +9,7 @@ from .calculator import Calculator
from .text_to_audio import TextToAudio
from .tesou import Tesou
from .fastchat import Fastchat
from .g2e import G2E
class BlackboxFactory:
@ -23,6 +24,7 @@ class BlackboxFactory:
self.tesou = Tesou()
self.fastchat = Fastchat()
self.audio_chat = AudioChat(self.asr, self.tesou, self.tts)
self.g2e = G2E()
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)
@ -48,4 +50,6 @@ class BlackboxFactory:
return self.fastchat
if blackbox_name == "audio_chat":
return self.audio_chat
if blackbox_name == "g2e":
return self.g2e
raise ValueError("Invalid blockbox type")