FROM python:3.8

RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

ENV TZ=Asia/Shanghai

RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

RUN apt-get update && apt-get install -y build-essential ffmpeg parallel aria2 && apt-get clean

COPY ./requirements.txt /workspace/requirements.txt

WORKDIR /workspace

RUN pip3 install -r requirements.txt && pip3 install webrtcvad-wheels

WORKDIR /

COPY . /workspace

VOLUME [ "/datasets", "/workspace/synthesizer/saved_models/" ]

ENV DATASET_MIRROR=default FORCE_RETRAIN=false TRAIN_DATASETS=aidatatang_200zh\ magicdata\ aishell3\ data_aishell TRAIN_SKIP_EXISTING=true

EXPOSE 7860

RUN chmod -R 777 /workspace

WORKDIR /workspace

CMD ["python3", "web.py"]