Update Dockerfile

Dockerfile

@@ -1,18 +1,36 @@
 FROM python:3.10-slim
 
-# Lebih cepat & lebih kecil
+# tools minimal
 RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
-# Install deps
-RUN pip install --no-cache-dir fastapi==0.110.3 uvicorn==0.30.1 rembg==2.0.57 pillow==10.3.0
+# lib python
+RUN pip install --no-cache-dir \
+    fastapi==0.110.3 \
+    uvicorn==0.30.1 \
+    rembg==2.0.57 \
+    pillow==10.3.0 \
+    python-multipart==0.0.9
 
-RUN pip install fastapi uvicorn rembg python-multipart
-
-# ---- FIX: nonaktifkan Numba JIT untuk hindari error caching ----
+# ---- FIX: numba & cache lokasi model ----
 ENV NUMBA_DISABLE_JIT=1
+ENV HOME=/root
+ENV U2NET_HOME=/root/.u2net
+ENV POOCH_HOME=/root/.cache/pooch
+
+# siapkan folder cache yg writable
+RUN mkdir -p /root/.u2net /root/.cache/pooch && chmod -R 777 /root/.u2net /root/.cache
 
 WORKDIR /app
 COPY app.py /app/app.py
 
+# (OPSI KUAT DISARANKAN) Prefetch model saat build agar request pertama cepat
+# ini akan otomatis download model U²Net ke U2NET_HOME/POOCH_HOME
+RUN python - <<'PY'
+from rembg.session_base import new_session
+# model default 'u2net' cukup untuk foto produk
+new_session('u2net')
+print('Model U2Net terunduh.')
+PY
+
 EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]