Update Dockerfile

Dockerfile

@@ -2,35 +2,21 @@ FROM python:3.10
 
 WORKDIR /app
 
-# =========================
-# 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/*
 
-# =========================
-# PYTHON DEPENDENCIES
-# =========================
 COPY requirements.txt .
+
 RUN pip install --no-cache-dir -r requirements.txt
 
-# =========================
-# COPY ENTIRE PROJECT (IMPORTANT FIX)
-# =========================
-COPY . .
+COPY app.py .
+
+COPY densenet121.pth .
+COPY densenet169.pth .
+COPY densenet201.pth .
+COPY cnn3d.pth .
 
-# =========================
-# PORT FOR HUGGING FACE
-# =========================
 EXPOSE 7860
 
-# =========================
-# START APP
-# =========================
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]