Update Dockerfile

Dockerfile

@@ -10,12 +10,14 @@ COPY . /app/
 # Install dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Set environment variables for Matplotlib and cache
+# Set environment variables for Matplotlib, cache, and rembg models
 ENV MPLCONFIGDIR=/tmp/matplotlib
 ENV XDG_CACHE_HOME=/tmp
+ENV U2NET_HOME=/tmp/u2net  # Ensure rembg can download models to a writable directory
 
-# Ensure the directories exist and set correct permissions
-RUN mkdir -p /tmp/matplotlib /tmp/numba_cache && chmod -R 777 /tmp/matplotlib /tmp/numba_cache
+# Ensure necessary directories exist and have the right permissions
+RUN mkdir -p /tmp/matplotlib /tmp/numba_cache /tmp/u2net && \
+    chmod -R 777 /tmp/matplotlib /tmp/numba_cache /tmp/u2net
 
 # Expose port 7860 (Hugging Face default)
 EXPOSE 7860