File size: 520 Bytes
50bfe7a
 
 
 
d0ff1b1
 
 
 
 
 
c9fdad1
46a0585
76e0b07
30b7d9a
d35479f
424839d
d35479f
50bfe7a
 
 
 
c9fdad1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
FROM python:3.10

WORKDIR /app

ENV HF_HOME=/app/hf_cache
ENV TRANSFORMERS_CACHE=/app/hf_cache
ENV DIFFUSERS_CACHE=/app/hf_cache

RUN mkdir -p /app/hf_cache

RUN pip install --no-cache-dir torch torchvision torchaudio diffusers fastapi uvicorn
RUN pip install git+https://github.com/huggingface/diffusers.git
RUN pip install transformers
RUN pip install accelerate
RUN pip install huggingface_hub
RUN pip install python-dotenv

COPY . /app

EXPOSE 7860

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