Corrected reference to sentence transformers dependency. Updated Dockerfile packages

Dockerfile

@@ -15,11 +15,11 @@ WORKDIR /src
 # Copy requirements file and install dependencies. Sentence transformers and Bertopic are installed without dependencies so that torch is not reinstalled.
 COPY requirements_aws.txt .
 
-RUN pip install torch==2.5.1+cpu --target=/install --index-url https://download.pytorch.org/whl/cpu \
-&& pip install --no-cache-dir --target=/install sentence-transformers==3.2.0 --no-deps \
-&& pip install --no-cache-dir --target=/install bertopic==0.16.4 --no-deps \
+RUN pip install torch>=2.6.0+cpu --target=/install --index-url https://download.pytorch.org/whl/cpu \
+&& pip install --no-cache-dir --target=/install sentence-transformers>=5.1.2 --no-deps \
+&& pip install --no-cache-dir --target=/install bertopic==0.17.3 --no-deps \
 && pip install --no-cache-dir --target=/install -r requirements_aws.txt \
-&& pip install --no-cache-dir --target=/install gradio==5.6.0
+&& pip install --no-cache-dir --target=/install gradio==5.49.1
 
 # Add /install to the PYTHONPATH
 ENV PYTHONPATH="/install:${PYTHONPATH}"

requirements.txt

@@ -16,7 +16,7 @@ presidio_analyzer>=2.2.360
 presidio_anonymizer>=2.2.360
 scipy>=1.15.3
 polars>=1.34.0
-sentence-transformers==5.2.0
+sentence-transformers==5.1.2
 torch>=2.6.0 --extra-index-url https://download.pytorch.org/whl/cu124
 #https://github.com/abetlen/llama-cpp-python/releases/download/v0.3.2/llama_cpp_python-0.3.2-cp311-cp311-win_amd64.whl # Exact wheel specified for windows
 #llama-cpp-python==0.3.2 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu

requirements_aws.txt

@@ -16,6 +16,6 @@ polars>=1.34.0
 transformers==4.57.1
 accelerate==1.11.0
 bertopic==0.17.3
-sentence-transformers==5.2.0
+sentence-transformers==5.1.2
 spaces==0.42.1
 numpy==2.2.6

requirements_gpu.txt

@@ -18,7 +18,7 @@ presidio_anonymizer>=2.2.360
 scipy>=1.15.3
 polars>=1.34.0
 llama-cpp-python==0.3.4 --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu124
-sentence-transformers==5.2.0
+sentence-transformers==5.1.2
 spaces==0.42.1
 numpy==2.2.6