FROM git.modelhub.org.cn:9443/enginex-ascend/cann:8.2.rc1-910b-ubuntu22.04-py3.11

WORKDIR /root

COPY requirements.txt /root
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

# RUN apt update && apt install -y vim net-tools wget
# RUN wget https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp311-cp311-manylinux_2_28_aarch64.whl
ADD torch-2.6.0+cpu-cp311-cp311-manylinux_2_28_aarch64.whl /root/
ADD torch_npu-2.6.0-cp311-cp311-manylinux_2_28_aarch64.whl /root/
RUN pip install torch-2.6.0+cpu-cp311-cp311-manylinux_2_28_aarch64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN pip install torch_npu-2.6.0-cp311-cp311-manylinux_2_28_aarch64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
RUN pip install torchaudio==2.6.0 funasr==1.2.6 openai-whisper -i https://pypi.tuna.tsinghua.edu.cn/simple

ADD . /root/
ADD nltk_data.tar.gz /root/
RUN tar -xvzf nltk_data.tar.gz

RUN cp ./replaced_files/ascend910b/auto_model.py /usr/local/python3.11.13/lib/python3.11/site-packages/funasr/auto/

EXPOSE 80
ENTRYPOINT ["bash"]
CMD ["./start_funasr.sh"]