Dockerfile

FROM python:3.10.9

WORKDIR /code

ENV TRANSFORMERS_CACHE=/tmp/cache/huggingface
ENV HF_HOME=/tmp/huggingface-cache

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

RUN mkdir -p /tmp/cache/huggingface && chmod -R 777 /tmp/cache/huggingface
RUN mkdir -p /tmp/huggingface-cache && chmod -R 777 /tmp/huggingface-cache

RUN python3 -m pip install -U -I --no-deps --no-cache-dir https://github.com/mikf/gallery-dl/archive/master.tar.gz
RUN python3 -m pip install -U --no-cache-dir yt-dlp
RUN apt-get update && apt-get install -y ffmpeg
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

COPY . .

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