trying out a fix

Dockerfile

@@ -2,37 +2,39 @@ FROM nvcr.io/nvidia/tensorflow:20.12-tf1-py3
 
 USER root
 
-# Fix GPG Keys and install only necessary system graphics/glib 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 \
-    libglib2.0-0 \
+    && apt-get update && apt-get install -y libgl1-mesa-glx libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Set up user and permissions
+# FORCE Python 3.7 as the only 'python' and 'python3'
+# This stops the system from accidentally using 3.8
+RUN ln -sf /usr/bin/python3.7 /usr/bin/python3 && \
+    ln -sf /usr/bin/python3.7 /usr/bin/python && \
+    ln -sf /usr/bin/python3.7 /usr/local/bin/python
+
+# Set up user
 RUN useradd -m -u 1000 user
 RUN mkdir -p /home/user/app && chown -R user:user /home/user/app
 
-# NVIDIA's Python 3.7 is typically at /usr/local/bin/python3
-# We create a symlink so 'python3.7' command works specifically
-RUN ln -s $(which python3) /usr/local/bin/python3.7
-
-# Force paths
+# Strict Environment Paths
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    PYTHONPATH=/usr/local/lib/python3.7/dist-packages \
+    PATH=/home/user/.local/bin:/usr/local/bin:/usr/bin:/sbin:/bin \
+    PYTHONPATH=/usr/local/lib/python3.7/dist-packages:/home/user/app \
     HF_HOME=/home/user/app/.cache
 
 WORKDIR /home/user/app
 
-# Now use the symlink we just created
-RUN python3.7 -m pip install --no-cache-dir --upgrade "pip<23.0"
+# Upgrade pip for 3.7 specifically
+RUN python -m pip install --no-cache-dir --upgrade "pip<23.0" "setuptools<60"
 
+# Install requirements
 COPY requirements.txt .
-RUN python3.7 -m pip install --no-cache-dir -r requirements.txt
+RUN python -m pip install --no-cache-dir -r requirements.txt
 
 COPY --chown=user:user . .
 
 USER user
 
-CMD ["python3.7", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Using 'python -m uvicorn' ensures we use the version linked above
+CMD ["python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,14 +1,21 @@
-try:
-    import keras
-except ImportError:
-    from tensorflow import keras
-    import sys
-    sys.modules['keras'] = keras 
 
 import os
 import io
 import numpy as np
 import tensorflow as tf
+import sys 
+
+try:
+    import keras
+except ImportError:
+    from tensorflow import keras
+
+import keras.engine as KE 
+sys.modules['keras.engine'] = keras.engine
+sys.modules['keras.layers'] = keras.layers
+sys.modules['keras.models'] = keras.models
+sys.modules['keras.utils'] = keras.utils
+
 from fastapi import FastAPI, File, UploadFile
 from fastapi.middleware.cors import CORSMiddleware 
 from PIL import Image