File size: 551 Bytes
3007746
 
 
 
 
 
 
360f55a
 
 
 
 
 
 
 
 
3007746
 
360f55a
3007746
02d730b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
FROM python:3.9

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
ENV HF_HOME=/code/transformers_cache
ENV TRANSFORMERS_CACHE=/code/transformers_cache
ENV HF_DATASETS_CACHE=/code/datasets_cache
ENV TOKENIZERS_PARALLELISM=true

VOLUME /code/transformers_cache
VOLUME /code/datasets_cache
VOLUME /code/app
VOLUME /code/models

COPY . .
RUN transformers-cli download succinctly/text2image-prompt-generator

CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]