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