From ee84580ad323d9878883eacd9e80549d7f322d57 Mon Sep 17 00:00:00 2001
From: Sora <654163754@qq.com>
Date: Wed, 13 Dec 2023 11:08:01 +0800
Subject: [PATCH] feat: update fastapi.py. support infer 2.2 models

---
 server_fastapi.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/server_fastapi.py b/server_fastapi.py
index 84e8765..e9c1678 100644
--- a/server_fastapi.py
+++ b/server_fastapi.py
@@ -201,7 +201,7 @@ if __name__ == "__main__":
         language: str,
         auto_translate: bool,
         auto_split: bool,
-        emotion: Optional[int] = None,
+        emotion: Optional[Union[int, str]] = None,
         reference_audio=None,
     ) -> Union[Response, Dict[str, any]]:
         """TTS实现函数"""
@@ -291,7 +291,7 @@ if __name__ == "__main__":
         language: str = Query(None, description="语言"),  # 若不指定使用语言则使用默认值
         auto_translate: bool = Query(False, description="自动翻译"),
         auto_split: bool = Query(False, description="自动切分"),
-        emotion: Optional[int] = Query(None, description="emo"),
+        emotion: Optional[Union[int, str]] = Query(None, description="emo"),
         reference_audio: UploadFile = File(None),
     ):
         """语音接口，若需要上传参考音频请仅使用post请求"""
@@ -330,7 +330,7 @@ if __name__ == "__main__":
         language: str = Query(None, description="语言"),  # 若不指定使用语言则使用默认值
         auto_translate: bool = Query(False, description="自动翻译"),
         auto_split: bool = Query(False, description="自动切分"),
-        emotion: Optional[int] = Query(None, description="emo"),
+        emotion: Optional[Union[int, str]] = Query(None, description="emo"),
     ):
         """语音接口"""
         logger.info(