mac9087 commited on
Commit
19a8ebb
·
verified ·
1 Parent(s): cfa68ff

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -1
Dockerfile CHANGED
@@ -33,7 +33,11 @@ COPY requirements.txt .
33
  RUN pip install --no-cache-dir -U pip && \
34
  pip install --no-cache-dir -r requirements.txt
35
 
36
- # Copy application code
 
 
 
 
37
  COPY app.py .
38
 
39
  # Expose the port the app runs on (standard for Hugging Face Spaces)
 
33
  RUN pip install --no-cache-dir -U pip && \
34
  pip install --no-cache-dir -r requirements.txt
35
 
36
+ # Download U^2-Net model file
37
+ RUN wget -O u2net.pth https://github.com/xuebinqin/U-2-Net/raw/master/model/u2net.pth
38
+
39
+ # Copy U^2-Net module and application code
40
+ COPY u2net.py .
41
  COPY app.py .
42
 
43
  # Expose the port the app runs on (standard for Hugging Face Spaces)