diff --git a/app.py b/app.py index 1dc50dc..1d65e91 100644 --- a/app.py +++ b/app.py @@ -17,7 +17,8 @@ def inference(text): print(outputs[0]) last_hidden_states = outputs.last_hidden_state - return outputs[0] + result = tokenizer.decode(outputs[0], skip_special_tokens=True) + return result examples=[["Studies have been shown that owning a dog is good for you"]] diff --git a/build.sh b/build.sh index bb837f6..f7d0cef 100644 --- a/build.sh +++ b/build.sh @@ -1,6 +1,6 @@ export release_name="translation-t5-large" export image="artifacts.iflytek.com/docker-private/atp/aicloud/translation" -export version="493bdb2a46" +export version="0b3e5f9bae" export gpu=true helm upgrade -n default --install $release_name --set image.repository="$image" \ --set image.tag="$version" --set image.pullPolicy=Always --set service.port=7860 \