flakego commited on
Commit
512bc1a
·
verified ·
1 Parent(s): e4cf2d8

Add automatic Doubao mask detection with fast and quality modes

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -2
Dockerfile CHANGED
@@ -4,6 +4,7 @@ ENV PYTHONUNBUFFERED=1 \
4
  PIP_NO_CACHE_DIR=1 \
5
  LAMA_MODEL_REPO=Carve/LaMa-ONNX \
6
  LAMA_MODEL_FILE=lama_fp32.onnx \
 
7
  LOAD_MODEL_ON_STARTUP=1
8
 
9
  WORKDIR /app
@@ -15,9 +16,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
15
  && rm -rf /var/lib/apt/lists/*
16
 
17
  COPY requirements.txt .
18
- RUN pip install --upgrade pip && pip install -r requirements.txt
 
 
19
 
20
- COPY app.py .
21
 
22
  EXPOSE 7860
23
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
4
  PIP_NO_CACHE_DIR=1 \
5
  LAMA_MODEL_REPO=Carve/LaMa-ONNX \
6
  LAMA_MODEL_FILE=lama_fp32.onnx \
7
+ DEFAULT_MODE=fast \
8
  LOAD_MODEL_ON_STARTUP=1
9
 
10
  WORKDIR /app
 
16
  && rm -rf /var/lib/apt/lists/*
17
 
18
  COPY requirements.txt .
19
+ RUN pip install --upgrade pip \
20
+ && pip install torch --index-url https://download.pytorch.org/whl/cpu \
21
+ && pip install -r requirements.txt
22
 
23
+ COPY app.py doubao_alpha.png ./
24
 
25
  EXPOSE 7860
26
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]