Spaces:
Sleeping
Sleeping
File size: 684 Bytes
54e8ea7 5ac2785 54e8ea7 5ac2785 54e8ea7 5ac2785 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
FROM python:3.10-slim
WORKDIR /code
COPY . .
# Install all dependencies (including pillow)
RUN pip install --upgrade pip && pip install -r requirements.txt
# Set cache dir env vars (fixes Hugging Face Space permissions)
ENV HF_HOME="/data"
ENV TRANSFORMERS_CACHE="/data"
RUN mkdir -p /data
# (Optional, but speeds up runtime) Pre-download BLIP model weights
RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"
EXPOSE 7860
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
|