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