diff --git a/beit_base_patch16_224_pt22k_ft22k/Dockerfile b/beit_base_patch16_224_pt22k_ft22k/Dockerfile index 2f124b8..1a9c1e3 100644 --- a/beit_base_patch16_224_pt22k_ft22k/Dockerfile +++ b/beit_base_patch16_224_pt22k_ft22k/Dockerfile @@ -2,6 +2,10 @@ FROM public.ecr.aws/iflytek-open/cuda-go-python-base:11.6-1.17-3.9.13-ubuntu1804 +RUN sed -i 's@//.*archive.ubuntu.com@//mirrors.ustc.edu.cn@g' /etc/apt/sources.list +RUN sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list + +RUN --mount=target=/root/packages.txt,source=packages.txt apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/* WORKDIR /home/user/app RUN useradd -m -u 1000 user @@ -14,8 +18,11 @@ RUN pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/ RUN pip install --no-cache-dir pip==22.3.1 RUN --mount=target=requirements.txt,source=requirements.txt pip install --no-cache-dir -r requirements.txt - COPY --chown=1000 ./ /home/user/app +RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash +RUN apt-get install git-lfs +RUN git lfs install +RUN git http://172.16.59.16:3000/microsoft/beit-base-patch16-224-pt22k-ft22k.git CMD ["python3", "app.py"] \ No newline at end of file diff --git a/beit_base_patch16_224_pt22k_ft22k/app.py b/beit_base_patch16_224_pt22k_ft22k/app.py index 0e1db1c..04495dc 100644 --- a/beit_base_patch16_224_pt22k_ft22k/app.py +++ b/beit_base_patch16_224_pt22k_ft22k/app.py @@ -19,7 +19,7 @@ def inference(img): title = "Image classification:beit-base-patch16-224-pt22k-ft22k" -description = "Gradio Demo for beit-base-patch16-224-pt22k-ft22k. To use it, simply upload your image, or click one of the examples to load them." +description = "这是beit-base-patch16-224-pt22k-ft22k的Gradio Demo。用于图像分类。上传你想要的图像或者点击下面的示例来加载它。" article = "

Github Repo Pytorch

visitor badge

" examples = [['example_cat.jpg'], ['Masahiro.png']] @@ -32,5 +32,5 @@ demo = gr.Interface( article=article, examples=examples) -demo.launch() +demo.launch(server_name="0.0.0.0") diff --git a/beit_base_patch16_224_pt22k_ft22k/beit-base-patch16-224-pt22k-ft22k b/beit_base_patch16_224_pt22k_ft22k/beit-base-patch16-224-pt22k-ft22k deleted file mode 160000 index 9da3011..0000000 --- a/beit_base_patch16_224_pt22k_ft22k/beit-base-patch16-224-pt22k-ft22k +++ /dev/null @@ -1 +0,0 @@ -Subproject commit 9da301148150e37e533abef672062fa49f6bda4f diff --git a/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmp4d3jvm17.jpg b/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmp4d3jvm17.jpg deleted file mode 100644 index e131e8e..0000000 Binary files a/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmp4d3jvm17.jpg and /dev/null differ diff --git a/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmpbfwv5bbb.png b/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmpbfwv5bbb.png deleted file mode 100644 index 1a43102..0000000 Binary files a/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmpbfwv5bbb.png and /dev/null differ diff --git a/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmpcm8vuiq9.png b/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmpcm8vuiq9.png deleted file mode 100644 index 1a43102..0000000 Binary files a/beit_base_patch16_224_pt22k_ft22k/flagged/img/tmpcm8vuiq9.png and /dev/null differ diff --git a/beit_base_patch16_224_pt22k_ft22k/flagged/log.csv b/beit_base_patch16_224_pt22k_ft22k/flagged/log.csv deleted file mode 100644 index 60e813b..0000000 --- a/beit_base_patch16_224_pt22k_ft22k/flagged/log.csv +++ /dev/null @@ -1,4 +0,0 @@ -img,output,flag,username,timestamp -D:\workspace\py\image_classification\beit_base_patch16_224_pt22k_ft22k\flagged\img\tmpcm8vuiq9.png,,,,2023-03-30 10:03:06.225359 -D:\workspace\py\image_classification\beit_base_patch16_224_pt22k_ft22k\flagged\img\tmpbfwv5bbb.png,,,,2023-03-30 10:03:08.489673 -D:\workspace\py\image_classification\beit_base_patch16_224_pt22k_ft22k\flagged\img\tmp4d3jvm17.jpg,,,,2023-03-30 10:03:11.483544 diff --git a/beit_base_patch16_224_pt22k_ft22k/packages.txt b/beit_base_patch16_224_pt22k_ft22k/packages.txt new file mode 100644 index 0000000..b399c93 --- /dev/null +++ b/beit_base_patch16_224_pt22k_ft22k/packages.txt @@ -0,0 +1,2 @@ +git +curl \ No newline at end of file diff --git a/beit_base_patch16_224_pt22k_ft22k/requirements.txt b/beit_base_patch16_224_pt22k_ft22k/requirements.txt index 973b15c..5e5bf22 100644 --- a/beit_base_patch16_224_pt22k_ft22k/requirements.txt +++ b/beit_base_patch16_224_pt22k_ft22k/requirements.txt @@ -1,2 +1,3 @@ gradio==3.21.0 transformers==4.27.1 +torch==2.0.0 \ No newline at end of file