tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

feat: update fastapi.py. support infer 2.2 models

Browse Source
This commit is contained in:
Sora
2023-12-13 11:08:01 +08:00
parent 79efeecdf6
commit ee84580ad3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server_fastapi.py
View File

@@ -201,7 +201,7 @@ if __name__ == "__main__":
language: str, language: str,
auto_translate: bool, auto_translate: bool,
auto_split: bool, auto_split: bool,
emotion: Optional[int] = None, emotion: Optional[Union[int, str]] = None,
reference_audio=None, reference_audio=None,
) -> Union[Response, Dict[str, any]]: ) -> Union[Response, Dict[str, any]]:
"""TTS实现函数""" """TTS实现函数"""
@@ -291,7 +291,7 @@ if __name__ == "__main__":
language: str = Query(None, description="语言"), # 若不指定使用语言则使用默认值 language: str = Query(None, description="语言"), # 若不指定使用语言则使用默认值
auto_translate: bool = Query(False, description="自动翻译"), auto_translate: bool = Query(False, description="自动翻译"),
auto_split: 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), reference_audio: UploadFile = File(None),
): ):
"""语音接口，若需要上传参考音频请仅使用post请求""" """语音接口，若需要上传参考音频请仅使用post请求"""
@@ -330,7 +330,7 @@ if __name__ == "__main__":
language: str = Query(None, description="语言"), # 若不指定使用语言则使用默认值 language: str = Query(None, description="语言"), # 若不指定使用语言则使用默认值
auto_translate: bool = Query(False, description="自动翻译"), auto_translate: bool = Query(False, description="自动翻译"),
auto_split: 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( logger.info(