Updated Dockerfile

Dockerfile

@@ -1,20 +1,50 @@
-# Use Python 3.12 slim image as base
-FROM python:3.12-slim
+# Use Red Hat Universal Base Image (UBI) 9 with Python 3.11
+FROM registry.access.redhat.com/ubi9/python-311:latest
 
+# Switch to root for system installations
+USER 0
 
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
+# Install system dependencies
+RUN dnf install -y \
+    gcc \
+    gcc-c++ \
+    git \
+    make \
+    && dnf clean all \
+    && rm -rf /var/cache/dnf
 
+# Create application user (using Red Hat standard practices)
+RUN useradd -u 1001 -g 0 -M -d /app user && \
+    mkdir -p /app && \
+    chown -R 1001:0 /app && \
+    chmod -R g=u /app
+
+# Set working directory
 WORKDIR /app
 
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+# Switch to non-root user
+USER 1001
+
+# Copy requirements first to leverage Docker cache
+COPY --chown=1001:0 ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Copy application code
+COPY --chown=1001:0 . /app
+COPY --chown=1001:0 main/ /app/main
+
+# Create directory for models cache
+RUN mkdir -p /app/.cache/huggingface && \
+    chmod -R g=u /app/.cache
 
-COPY --chown=user . /app
-COPY --chown=user main/ /app/main
+# Environment variables
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV HOME=/app
+ENV HUGGINGFACE_HOME=/app/.cache/huggingface
 
 EXPOSE 7860
 
 # Command to run the application
-CMD ["uvicorn", "main.main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main.main:app", "--host", "0.0.0.0", "--port", "7860"]

main/main.py

@@ -73,7 +73,8 @@ def create_app():
     # Launch inference workers (assuming single uvicorn worker for now)
     _MANAGER, _WORKER_PROCESSES = server.launch_inference_worker(num_uvicorn_servers=1)
 
-    # Get the FastAPI app
+    # Get the FastAPI appls
+
     app = server.app
 
     # Add CORS middleware