FROM corex:4.3.8

WORKDIR /root

ADD . /root/

COPY requirements.txt /root
RUN pip install -r requirements.txt -i https://nexus.4pd.io/repository/pypi-all/simple
# 安装torch是为了提供cuda库环境
RUN pip install transformers==4.51.3 -i https://nexus.4pd.io/repository/pypi-all/simple

ENTRYPOINT ["python3"]
CMD ["./main_qa.py"]