refactor dockerfile

Dockerfile

@@ -1,7 +1,35 @@
 # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.11.11
+# Stage 1: Build dependencies
+FROM python:3.11-slim AS build
+
+# Install system dependencies (pillow, torch dll butuh ini)
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    libjpeg-dev \
+    zlib1g-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# Buat user non-root
+RUN useradd -m -u 1000 user
+
+# Set workdir & salin requirements
+WORKDIR /app
+COPY --chown=user requirements.txt .
+
+# Install pip packages secara lokal (non-root)
+RUN pip install --no-cache-dir --user -r requirements.txt
+
+---
+
+# Stage 2: Final image (lebih ringan)
+FROM python:3.11-slim
+
+RUN apt-get update && apt-get install -y \
+    libjpeg-dev \
+    zlib1g-dev \
+    && rm -rf /var/lib/apt/lists/*
 
 RUN useradd -m -u 1000 user
 USER user
@@ -9,9 +37,11 @@ ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+# Copy pip packages hasil install dari stage build
+COPY --from=build /home/user/.local /home/user/.local
 
-COPY --chown=user . /app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Copy file kode app kamu
+COPY --chown=user . .
 
+EXPOSE 7860
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]