diff --git a/build.sh b/build.sh index f88a106..bb837f6 100644 --- a/build.sh +++ b/build.sh @@ -1,6 +1,6 @@ -export release_name="image-classification-beit-base" -export image="artifacts.iflytek.com/docker-private/atp/aicloud/image_classification" -export version="e68f0f3459" +export release_name="translation-t5-large" +export image="artifacts.iflytek.com/docker-private/atp/aicloud/translation" +export version="493bdb2a46" 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 \ diff --git a/requirements.txt b/requirements.txt index 5e5bf22..03b39f0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ gradio==3.21.0 transformers==4.27.1 -torch==2.0.0 \ No newline at end of file +torch==2.0.0 +sentencepiece