FROM git.modelhub.org.cn:9443/enginex-kunlunxin/diffuser:r200-8f

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"]
                                                                                                             
                                                                          

