Update Dockerfile

Dockerfile

@@ -4,13 +4,19 @@ FROM python:3.9-slim
 RUN apt-get update && apt-get install -y \
     libgl1 \
     libglib2.0-0 \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Set environment variables for Numba
+# Set environment variables
 ENV NUMBA_CACHE_DIR=/tmp/numba_cache \
-    NUMBA_DISABLE_CACHE=1
+    NUMBA_DISABLE_CACHE=1 \
+    U2NET_HOME=/app/models
+
+# Pre-download model during build
+RUN mkdir -p /app/models \
+    && wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx -P /app/models
 
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
@@ -18,4 +24,5 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY app.py .
 
 EXPOSE 5000
-CMD ["gunicorn", "--bind", "0.0.0.0:5000", "--timeout", "120", "app:app"]
+# Increase timeout to 5 minutes (300 seconds)
+CMD ["gunicorn", "--bind", "0.0.0.0:5000", "--timeout", "300", "--preload", "app:app"]