Update Dockerfile

Dockerfile

@@ -2,25 +2,35 @@ FROM python:3.10
 
 WORKDIR /app
 
-# install system deps
+# =========================
+# SYSTEM DEPENDENCIES
+# =========================
 RUN apt-get update && apt-get install -y \
     git \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libgl1 \
     && rm -rf /var/lib/apt/lists/*
 
-# install python deps
+# =========================
+# PYTHON DEPENDENCIES
+# =========================
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# copy app
-COPY app.py .
+# =========================
+# COPY ENTIRE PROJECT (IMPORTANT FIX)
+# =========================
+COPY . .
 
-# copy models (VERY IMPORTANT)
-COPY alzheimers_densenet121.pth .
-COPY alzheimers_densenet169.pth .
-COPY alzheimers_densenet201.pth .
-
-# expose port
+# =========================
+# PORT FOR HUGGING FACE
+# =========================
 EXPOSE 7860
 
-# run fastapi
+# =========================
+# START APP
+# =========================
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]