Update Dockerfile

Dockerfile

@@ -20,7 +20,7 @@ WORKDIR /app
 # Install Python dependencies
 RUN pip install --upgrade pip
 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-RUN pip install transformers accelerate requests psutil termcolor
+RUN pip install transformers accelerate requests psutil termcolor sklearn
 
 # Copy the current directory contents into the container at /app
 COPY . /app