Spaces:
Running
Running
Update Dockerfile
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -1,6 +1,8 @@
|
|
| 1 |
# Minimal Dockerfile for Hugging Face Spaces or local deployment
|
| 2 |
FROM python:3.10-slim
|
| 3 |
WORKDIR /code
|
|
|
|
|
|
|
| 4 |
COPY . .
|
| 5 |
RUN pip install --upgrade pip && pip install -r requirements.txt
|
| 6 |
EXPOSE 7860
|
|
|
|
| 1 |
# Minimal Dockerfile for Hugging Face Spaces or local deployment
|
| 2 |
FROM python:3.10-slim
|
| 3 |
WORKDIR /code
|
| 4 |
+
RUN pip install transformers torch
|
| 5 |
+
RUN python -c "from transformers import BlipProcessor, BlipForConditionalGeneration; BlipProcessor.from_pretrained('Salesforce/blip-image-captioning-base'); BlipForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-base')"
|
| 6 |
COPY . .
|
| 7 |
RUN pip install --upgrade pip && pip install -r requirements.txt
|
| 8 |
EXPOSE 7860
|