From 9ae57d4a24da418e188ce0e1fa00b2ab690fad8e Mon Sep 17 00:00:00 2001
From: SOULOFCINDER <841135647@qq.com>
Date: Thu, 6 Apr 2023 16:29:13 +0800
Subject: [PATCH] update

---
 distilgpt2/Dockerfile   | 9 ++++++++-
 distilgpt2/app.py       | 2 +-
 distilgpt2/packages.txt | 2 ++
 3 files changed, 11 insertions(+), 2 deletions(-)
 create mode 100644 distilgpt2/packages.txt

diff --git a/distilgpt2/Dockerfile b/distilgpt2/Dockerfile
index 2f124b8..bd7ff08 100644
--- a/distilgpt2/Dockerfile
+++ b/distilgpt2/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 clone http://172.16.59.16:3000/huggingface/distilgpt2.git
 
 CMD ["python3", "app.py"]
\ No newline at end of file
diff --git a/distilgpt2/app.py b/distilgpt2/app.py
index 89ac029..8dc9616 100644
--- a/distilgpt2/app.py
+++ b/distilgpt2/app.py
@@ -22,7 +22,7 @@ with gr.Blocks() as demo:
         """
     # Text generation:distilgpt2
     DistilGPT2（Distilled-GPT2）是一种英语模型，在GPT-2的监督下进行预训练。与 GPT-2 一样，DistilGPT2 可用于生成文本。
-    这是一个distilgpt2的Gradio Demo. 输入你想要的英文文本或者点击下面的示例文本来加载它.
+    这是一个distilgpt2的Gradio Demo。 输入你想要的英文文本或者点击下面的示例文本来加载它。
     """)
     with gr.Row():
         text_input = gr.Textbox()
diff --git a/distilgpt2/packages.txt b/distilgpt2/packages.txt
new file mode 100644
index 0000000..b399c93
--- /dev/null
+++ b/distilgpt2/packages.txt
@@ -0,0 +1,2 @@
+git
+curl
\ No newline at end of file