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 09:31:16 +00:00
parent e2106fc33c 335ff7cb8d
commit f57b669db2
107 changed files with 29145 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/blackbox/blackbox_factory.py
View File

@ -1,7 +1,8 @@
from .audio_chat import AudioChat
from .sum import SUM
from .sentiment import Sentiment
from .tts import TTS
from ..asr.asr import ASR
from .asr import ASR
from .audio_to_text import AudioToText
from .blackbox import Blackbox
from .calculator import Calculator
@ -13,14 +14,18 @@ class BlackboxFactor:
def __init__(self) -> None:
self.tts = TTS()
self.asr = ASR("./.env.yaml")
self.asr = ASR(".env.yaml")
self.sentiment = Sentiment()
self.sum = SUM()
self.calculator = Calculator()
self.audio_to_text = AudioToText()
self.text_to_audio = TextToAudio()
self.tesou = Tesou()
<<<<<<< HEAD
self.fastchat = Fastchat()
=======
self.audio_chat = AudioChat(self.asr, self.tesou, self.tts)
>>>>>>> refs/remotes/origin/main
def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs)
@ -42,6 +47,11 @@ class BlackboxFactor:
return self.sum
if blackbox_name == "tesou":
return self.tesou
<<<<<<< HEAD
if blackbox_name == "fastchat":
return self.fastchat
=======
if blackbox_name == "audio_chat":
return self.audio_chat
>>>>>>> refs/remotes/origin/main
raise ValueError("Invalid blockbox type")