From 8d2a4d0673717061380338b981c81175223cd788 Mon Sep 17 00:00:00 2001
From: Limbo <57550280+Alu-XXX@users.noreply.github.com>
Date: Mon, 13 May 2024 14:43:20 +0800
Subject: [PATCH] Update blackbox_factory.py

---
 src/blackbox/blackbox_factory.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/blackbox/blackbox_factory.py b/src/blackbox/blackbox_factory.py
index 345c9de..cd8b623 100644
--- a/src/blackbox/blackbox_factory.py
+++ b/src/blackbox/blackbox_factory.py
@@ -3,6 +3,7 @@ from .sentiment import Sentiment
 from .tts import TTS
 from .asr import ASR
 from .audio_to_text import AudioToText
+from .emotion import Emotion
 from .blackbox import Blackbox
 from .text_to_audio import TextToAudio
 from .tesou import Tesou
@@ -26,6 +27,7 @@ class BlackboxFactory:
                  asr: ASR,
                  tts: TTS, 
                  sentiment_engine: Sentiment, 
+                 emotion: Emotion,
                  tesou: Tesou, 
                  fastchat: Fastchat,
                  audio_chat: AudioChat,
@@ -41,6 +43,7 @@ class BlackboxFactory:
         self.models["tts"] = tts
         self.models["sentiment_engine"] = sentiment_engine
         self.models["tesou"] = tesou
+        self.models["emotion"] = emotion
         self.models["fastchat"] = fastchat
         self.models["audio_chat"] = audio_chat
         self.models["g2e"] = g2e
@@ -57,4 +60,4 @@ class BlackboxFactory:
         model = self.models.get(blackbox_name)
         if model is None:
             raise ValueError("Invalid blockbox type")
-        return model
\ No newline at end of file
+        return model