FROM mxc500-torch2.4-py310:mc2.33.0.6-ubuntu22.04-amd64

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

# 安装transformers 4.46.3
RUN /opt/conda/bin/python3 -m pip install --upgrade pip
RUN /opt/conda/bin/python3 -m pip install --no-cache-dir transformers==4.46.3
RUN /opt/conda/bin/python3 -m pip install flask==3.1.1


EXPOSE 80

ENTRYPOINT ["/opt/conda/bin/python3", "model_test_caltech_http_muxi.py"]