Update Dockerfile

Dockerfile

@@ -1,57 +1,63 @@
-# =========================================================
-# 1. BASE IMAGE (stable CUDA runtime)
-# =========================================================
-FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
-
-# =========================================================
-# 2. ENV VARIABLES
-# =========================================================
+FROM python:3.11-slim
+
+# ======================
+# ENV
+# ======================
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     PORT=7860
 
-# =========================================================
-# 3. SYSTEM DEPENDENCIES
-# =========================================================
+# ======================
+# SYSTEM DEPENDENCIES
+# ======================
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     build-essential \
+    libgl1 \
     libglib2.0-0 \
     libsm6 \
     libxrender1 \
     libxext6 \
+    && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
-# =========================================================
-# 4. WORK DIRECTORY
-# =========================================================
+# ======================
+# WORKDIR
+# ======================
 WORKDIR /app
 
-# =========================================================
-# 5. COPY REQUIREMENTS
-# =========================================================
+# ======================
+# COPY REQUIREMENTS
+# ======================
 COPY requirements.txt .
 
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
+# ======================
+# CLEAN INSTALL (IMPORTANT FIX)
+# ======================
+RUN pip install --no-cache-dir --upgrade pip
+
+# safety: remove broken installs if any base image has them
+RUN pip uninstall -y transformers || true
+
+RUN pip install --no-cache-dir -r requirements.txt
 
-# =========================================================
-# 6. COPY CODE
-# =========================================================
+# ======================
+# COPY CODE
+# ======================
 COPY . .
 
-# =========================================================
-# 7. NON-ROOT USER (SECURITY)
-# =========================================================
+# ======================
+# SECURITY USER
+# ======================
 RUN useradd -m appuser && chown -R appuser /app
 USER appuser
 
-# =========================================================
-# 8. EXPOSE PORT
-# =========================================================
+# ======================
+# PORT
+# ======================
 EXPOSE 7860
 
-# =========================================================
-# 9. RUN APP
-# =========================================================
+# ======================
+# RUN APP
+# ======================
 CMD ["sh", "-c", "uvicorn mainapp:app --host 0.0.0.0 --port ${PORT}"]