mac9087 commited on
Commit
08b7e09
·
verified ·
1 Parent(s): 5d3b2f3

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -33,8 +33,8 @@ COPY requirements.txt .
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 .
 
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 from Google Drive
37
+ RUN gdown https://drive.google.com/uc?id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ -O u2net.pth
38
 
39
  # Copy U^2-Net module and application code
40
  COPY u2net.py .