Update Dockerfile

Dockerfile

@@ -33,6 +33,7 @@ COPY ./requirements.txt /code/requirements.txt
 # Install Python packages in multiple steps to handle dependencies better
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir 'numpy==1.23.5' && \
+    pip install --no-cache-dir 'scikit-learn==1.6.1' && \
     pip install --no-cache-dir 'faiss-cpu==1.7.3' && \
     pip install --no-cache-dir einops>=0.6.1 && \
     pip install --no-cache-dir -r /code/requirements.txt