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