-
Install packages
Note: Use python3.8 and update pip in virtual enviroment
python3 -m venv env source env/bin/activate pip install -r requirements.txt
-
Create
server.json
for query (Only needed for Usage 2. and 3.)-
Under the repository directory
vim server.json
-
Inside server.json
{ "url": "your/server/url/for/query" }
-
- Deploy REST API
python src/summarizer.py
- Test the API
- Go through 1. first.
- Modify the query in
test_summarizer.py
or add any news you want as string without using query. - Then:
python test/test_summarizer.py
If: ModuleNotFoundError: No module named 'gql.transport.aiohttp'
Solution:pip uninstall gql pip install --pre gql[all]
- Test Pegasus
- Go through 1. first.
- Modify the query in
test_summarizer.py
or add any news you want as string without using query. - You can also change different pretrained Pegasus from Hugging Face.
- Then:
python test/test_pegasus.py
If: ModuleNotFoundError: No module named 'gql.transport.aiohttp'
Solution:pip uninstall gql pip install --pre gql[all]
@misc{zhang2019pegasus,
title={PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarization},
author={Jingqing Zhang and Yao Zhao and Mohammad Saleh and Peter J. Liu},
year={2019},
eprint={1912.08777},
archivePrefix={arXiv},
primaryClass={cs.CL}
}