StickerGen / Dockerfile
KarthikAI's picture
Update Dockerfile
1e6f971 verified
raw
history blame
531 Bytes
# Minimal Dockerfile for Hugging Face Spaces or local deployment
FROM python:3.10-slim
WORKDIR /code
RUN pip install transformers torch
RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"
COPY . .
RUN pip install --upgrade pip && pip install -r requirements.txt
EXPOSE 7860
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]