FROM mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310

WORKDIR /root

COPY requirements.txt /root

SHELL ["/bin/bash", "-c"]

RUN source /torch/venv3/pytorch/bin/activate && \ 
    pip uninstall -y deepspeed
RUN source /torch/venv3/pytorch/bin/activate && \ 
    pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

RUN source /torch/venv3/pytorch/bin/activate && \ 
    pip install funasr==1.2.6 openai-whisper -i https://pypi.tuna.tsinghua.edu.cn/simple

ADD . /root/
ADD nltk_data.tar.gz /root/
RUN tar -xzvf nltk_data.tar.gz
RUN cp ./replaced_files/cam_mlu370_x8/auto_model.py /torch/venv3/pytorch/lib/python3.10/site-packages/funasr/auto/
EXPOSE 80
ENTRYPOINT ["/bin/bash", "-c", "source /torch/venv3/pytorch/bin/activate && ./start_funasr.sh"]