vit-matte / Dockerfile
saicoder's picture
Add docker file
842dc01 unverified
Raw
History Blame Contribute Delete
362 Bytes
FROM python:3.11
WORKDIR /src
ADD ./ /src
RUN pip install --upgrade pip
RUN pip install --no-cache-dir -r requirements.txt
# Download Model
RUN python3 -c 'from app.cloth_segmentation.model import load_seg_model; load_seg_model(".checkpoint/model.pth")'
EXPOSE 80
CMD ["uvicorn", "app.server:app", "--proxy-headers", "--host", "0.0.0.0", "--port", "80"]