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

rebase

Browse Source
This commit is contained in:
superobk
2024-05-24 10:50:45 +08:00
parent 494ee31c5a
commit 68c453b219
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/blackbox/blackbox_factory.py
View File

@ -1,11 +1,11 @@
from blackbox.emotion import Emotion
from .chat import Chat
from .audio_chat import AudioChat from .audio_chat import AudioChat
from .sentiment import Sentiment from .sentiment import Sentiment
from .tts import TTS from .tts import TTS
from .asr import ASR from .asr import ASR
from .audio_to_text import AudioToText from .audio_to_text import AudioToText
from .blackbox import Blackbox from .blackbox import Blackbox
# from .text_to_audio import TextToAudio
# from .tesou import Tesou
from .fastchat import Fastchat from .fastchat import Fastchat
from .g2e import G2E from .g2e import G2E
from .text_and_image import TextAndImage from .text_and_image import TextAndImage
@ -26,24 +26,22 @@ class BlackboxFactory:
asr: ASR, asr: ASR,
tts: TTS, tts: TTS,
sentiment_engine: Sentiment, sentiment_engine: Sentiment,
#emotion: Emotion, emotion: Emotion,
fastchat: Fastchat, fastchat: Fastchat,
audio_chat: AudioChat, audio_chat: AudioChat,
g2e: G2E, g2e: G2E,
text_and_image: TextAndImage, text_and_image: TextAndImage,
#chroma_query: ChromaQuery,
#chroma_upsert: ChromaUpsert,
#chroma_chat: ChromaChat,
melotts: MeloTTS, melotts: MeloTTS,
vlms: VLMS, vlms: VLMS,
chroma_query: ChromaQuery, chroma_query: ChromaQuery,
chroma_upsert: ChromaUpsert, chroma_upsert: ChromaUpsert,
chroma_chat: ChromaChat) -> None: chroma_chat: ChromaChat,
chat: Chat) -> None:
self.models["audio_to_text"] = audio_to_text self.models["audio_to_text"] = audio_to_text
self.models["asr"] = asr self.models["asr"] = asr
self.models["tts"] = tts self.models["tts"] = tts
self.models["sentiment_engine"] = sentiment_engine self.models["sentiment_engine"] = sentiment_engine
#self.models["emotion"] = emotion self.models["emotion"] = emotion
self.models["fastchat"] = fastchat self.models["fastchat"] = fastchat
self.models["audio_chat"] = audio_chat self.models["audio_chat"] = audio_chat
self.models["g2e"] = g2e self.models["g2e"] = g2e
@ -51,6 +49,9 @@ class BlackboxFactory:
self.models["chroma_query"] = chroma_query self.models["chroma_query"] = chroma_query
self.models["chroma_upsert"] = chroma_upsert self.models["chroma_upsert"] = chroma_upsert
self.models["chroma_chat"] = chroma_chat self.models["chroma_chat"] = chroma_chat
self.models["melotts"] = melotts
self.models["vlms"] = vlms
self.models["chat"] = chat
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
return self.processing(*args, **kwargs) return self.processing(*args, **kwargs)