Update Dockerfile to install requirements earlier; add scikit-learn to requirements.txt

Dockerfile

@@ -9,12 +9,12 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt ./
+RUN pip3 install -r requirements.txt
+
 COPY src/ ./src/
 COPY input/ ./input/
 COPY models/ ./models/
 
-RUN pip3 install -r requirements.txt
-
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

requirements.txt

@@ -8,4 +8,5 @@ st-link-analysis
 torch
 tqdm
 transformers
-torch-geometric
+torch-geometric
+scikit-learn