Skip to content

Commit c4e5972

Browse files
committed
update interface return
1 parent cc73b5a commit c4e5972

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

docker-compose.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ services:
44
build:
55
context: .
66
dockerfile: Dockerfile
7-
image: paidax/funasr_python:0.2.3
7+
image: paidax/funasr_python:0.3
88
runtime: nvidia
99
ports:
1010
- 9527:9527

main.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,10 @@ async def predict(items: Audio):
118118
decoded_data = convert_audio_to_wav(decoded_data)
119119
load_model(model_type=items.model_type, hotword=items.hotword)
120120
rec_result = loaded_model["model"](audio_in=decoded_data)
121+
if items.model_type=='normal' or items.model_type=='long':
122+
rec_result = {
123+
"text": rec_result["text"]
124+
}
121125
log.info(rec_result)
122126
return rec_result
123127

0 commit comments

Comments
 (0)