chemistrymath commited on
Commit
810d904
·
verified ·
1 Parent(s): db7092f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -13,7 +13,8 @@ RUN pip install --no-cache-dir -r requirements.txt
13
  # Set environment variables for Matplotlib, cache, and rembg models
14
  ENV MPLCONFIGDIR=/tmp/matplotlib
15
  ENV XDG_CACHE_HOME=/tmp
16
- ENV U2NET_HOME=/tmp/u2net # Ensure rembg can download models to a writable directory
 
17
 
18
  # Ensure necessary directories exist and have the right permissions
19
  RUN mkdir -p /tmp/matplotlib /tmp/numba_cache /tmp/u2net && \
 
13
  # Set environment variables for Matplotlib, cache, and rembg models
14
  ENV MPLCONFIGDIR=/tmp/matplotlib
15
  ENV XDG_CACHE_HOME=/tmp
16
+ ENV U2NET_HOME=/tmp/u2net
17
+ # Ensure rembg can download models to a writable directory
18
 
19
  # Ensure necessary directories exist and have the right permissions
20
  RUN mkdir -p /tmp/matplotlib /tmp/numba_cache /tmp/u2net && \