updated docker

Dockerfile

@@ -2,19 +2,18 @@ FROM nvcr.io/nvidia/tensorflow:20.12-tf1-py3
 
 USER root
 
-# Fix GPG Keys
+# Fix GPG Keys and install only necessary system graphics/glib 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 \
     libglib2.0-0 \
-    python3.7-dev \
     && rm -rf /var/lib/apt/lists/*
 
 # Set up user and permissions
 RUN useradd -m -u 1000 user
 RUN mkdir -p /home/user/app && chown -R user:user /home/user/app
 
-# Force Python 3.7 paths
+# Force Python 3.7 paths and HF cache location
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=/usr/local/lib/python3.7/dist-packages \
@@ -31,4 +30,5 @@ COPY --chown=user:user . .
 
 USER user
 
+# Use python3.7 -m uvicorn to ensure we stay in the TF1.15 environment
 CMD ["python3.7", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]