FROM combricon-mlu370x4-base:v0.2.0-tgiv1.4.3-btv0.6.0-pt2.1-x86_64-ubuntu22.04-py310


RUN if [ ! -d "/torch/venv3/pytorch" ]; then echo "虚拟环境路径缺失！"; exit 1; fi


WORKDIR /workspace/
COPY ./model_test_caltech_http_mlu370.py /workspace/
COPY ./microsoft_beit_base_patch16_224_pt22k_ft22k /model


# 安装transformers 4.46.3
RUN pip install transformers==4.46.3 uvicorn\[standard\] fastapi flask==3.1.1

EXPOSE 80

ENTRYPOINT ["python3", "model_test_caltech_http_mlu370.py"]