FROM r200-8f_xmlir-ubuntu_2004_x86_64:v0.27

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


# 安装transformers 4.46.3
RUN /root/miniconda/envs/python38_torch201_cuda/bin/python3 -m pip install --upgrade pip
RUN /root/miniconda/envs/python38_torch201_cuda/bin/python3 -m pip install --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple transformers==4.46.3
RUN /root/miniconda/envs/python38_torch201_cuda/bin/python3 -m pip install flask

EXPOSE 80

ENTRYPOINT ["/root/miniconda/envs/python38_torch201_cuda/bin/python3", "model_test_caltech_http_kunlunxin.py"]