ai-lab / Dockerfile
ClemSummer's picture
Moved pth file to HF dataset
ac9c6c1
raw
history blame
395 Bytes
# 🐍 Use official Python
FROM python:3.11-slim
WORKDIR /app
COPY . .
RUN wget https://huggingface.co/datasets/ClemSummer/clip-checkpoints/resolve/main/CLIPEncoder_40epochs_unfreeze12.pth \
-O vit_captioning/artifacts/CLIPEncoder_40epochs_unfreeze12.pth
RUN pip install --upgrade pip
RUN pip install -r requirements.txt
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]