updated package versions

Dockerfile

@@ -2,7 +2,7 @@ FROM nvcr.io/nvidia/tensorflow:20.12-tf1-py3
 
 USER root
 
-# Fix GPG Keys for expired NVIDIA repositories and install system libs
+# Fix GPG Keys and install system libs
 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub \
     && apt-get update && apt-get install -y \
     libgl1-mesa-glx \
@@ -10,7 +10,7 @@ RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/
     python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Upgrade pip to a version compatible with Python 3.7
+# Upgrade pip but keep it compatible with Python 3.7
 RUN pip install --no-cache-dir --upgrade "pip<23.0"
 
 # Set up user
@@ -19,9 +19,11 @@ ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
 WORKDIR $HOME/app
 
+# Install requirements
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy app code
 COPY --chown=user:user . .
 
 USER user

requirements.txt

@@ -22,16 +22,14 @@ uvicorn
 python-multipart
 huggingface-hub
 pandas==1.3.5
-numpy==1.19.5          
+numpy==1.19.5
 matplotlib==3.5.3
 scikit-image==0.19.3
-opencv-python==4.5.5.64 
+opencv-python-headless==4.5.5.64
 h5py==2.10.0
-keras==2.2.4           
-tensorflow-gpu==1.15.0
 Keras-applications==1.0.7
-protobuf==3.19.6 
+protobuf==3.19.6
 seaborn==0.12.2
 scikit-learn==1.0.2
-scipy==1.7.3 
+scipy==1.7.3
 Pillow==9.5.0