diff --git a/docker_images/sentence_transformers/Dockerfile b/docker_images/sentence_transformers/Dockerfile index 4ea04f7b..12f436ed 100644 --- a/docker_images/sentence_transformers/Dockerfile +++ b/docker_images/sentence_transformers/Dockerfile @@ -1,10 +1,10 @@ -FROM tiangolo/uvicorn-gunicorn:python3.8 -LABEL maintainer="Omar " +FROM tiangolo/uvicorn-gunicorn:python3.10 +LABEL maintainer="Tom Aarsen " # Add any system dependency here # RUN apt-get update -y && apt-get install libXXX -y -RUN pip3 install --no-cache-dir torch==1.13.0 +RUN pip3 install --no-cache-dir torch==2.4.1 COPY ./requirements.txt /app diff --git a/docker_images/sentence_transformers/requirements.txt b/docker_images/sentence_transformers/requirements.txt index 686e5f2e..2fe6e6e1 100644 --- a/docker_images/sentence_transformers/requirements.txt +++ b/docker_images/sentence_transformers/requirements.txt @@ -1,9 +1,9 @@ starlette==0.27.0 api-inference-community==0.0.32 -sentence-transformers==3.0.1 -transformers==4.41.1 -tokenizers==0.19.1 +sentence-transformers==3.3.1 +transformers==4.48.0 +tokenizers==0.21.0 protobuf==3.18.3 -huggingface_hub==0.23.3 +huggingface_hub==0.27.1 sacremoses==0.0.53 # dummy. diff --git a/tests/test_audio.py b/tests/test_audio.py index f5fba0fc..266eddd4 100644 --- a/tests/test_audio.py +++ b/tests/test_audio.py @@ -36,7 +36,7 @@ def test_malformed_audio(self): def test_dual_channel(self): bpayload = self.read("sample1_dual.ogg") payload, params = normalize_payload_audio(bpayload, 16000) - self.assertEqual(payload.shape, (219520,)) + self.assertEqual(payload.shape, (219040,)) def test_original_webm(self): bpayload = self.read("sample1.webm")