From 4fa9bdfe5f19795f1261d829e86d0e2897f8c8dc Mon Sep 17 00:00:00 2001
From: jianjiang <jianjiang@iflytek.com>
Date: Fri, 21 Apr 2023 18:47:32 +0800
Subject: [PATCH] ok

---
 Dockerfile       | 2 +-
 app.py           | 2 ++
 requirements.txt | 1 +
 3 files changed, 4 insertions(+), 1 deletion(-)
 create mode 100644 requirements.txt

diff --git a/Dockerfile b/Dockerfile
index e6f7bb5..5b47bb2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,6 +6,6 @@ WORKDIR /app
 COPY . /app
 
 RUN pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple 
-#RUN pip install -r requirements.txt
+RUN pip install -r requirements.txt
 
 CMD ["python", "app.py"]
diff --git a/app.py b/app.py
index 3b3f504..4376709 100644
--- a/app.py
+++ b/app.py
@@ -15,6 +15,8 @@ theme = gr.themes.Default(radius_size=sizes.radius_none).set(
 )
 
 model_name = 'google/pegasus-cnn_dailymail'
+#model_name = '/mnt/atpdata/models_hub/huggingface/summarization/google/pegasus-cnn_dailymail'
+#model_name = '/models/pegasus-cnn_dailymail'
 summarization_model = pipeline('summarization', model=model_name, tokenizer=model_name, device=0 if torch.cuda.is_available() else -1)
 
 def generate_abstractive_summary(text, type, min_len=120, max_len=512, **kwargs):
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..89cfdc0
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+protobuf==3.20.1