File size: 693 Bytes
e83ff6c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a674d29
e83ff6c
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
FROM python:3.11-slim

RUN useradd -m -u 1000 user
WORKDIR /app

RUN apt-get update && apt-get install -y --no-install-recommends \

    libgl1 \
    libglib2.0-0 \
    curl \
    && rm -rf /var/lib/apt/lists/*

COPY requirements.txt .
RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu \

 && pip install --no-cache-dir -r requirements.txt

COPY . .

RUN chown -R user:user /app

ENV PATCH_MFFT_CHECKPOINT_DIR=/app/checkpoints \
    PATCH_MFFT_IMG_SIZE=384 \
    PATCH_MFFT_VARIANT=base \
    HF_HOME=/tmp/hf \
    PORT=7860

USER user
EXPOSE 7860
CMD ["uvicorn", "api.main:app", "--host", "0.0.0.0", "--port", "7860"]