Xingyu-Romantic/Text-Summarization-Based-On-Bart

PaddlePadlle, Bart, Text Summarization

自动文本摘要生成 🎉

2019 人工智能 芦星宇

File Structure 🎨

├── LICENSE
├── README.md
├── dataset
│   └── csl_title_public
│       ├── csl_title_dev.json
│       ├── csl_title_predict.json
│       ├── csl_title_test.json
│       └── csl_title_train.json
├── deploy
│   ├── demo.html
│   ├── deploy.py
│   ├── predict.py
│   └── static
│       ├── css
│       │   ├── base.css
│       │   ├── bootstrap-theme.min.css
│       │   └── bootstrap.min.css
│       └── templates
│           ├── common
│           │   └── base.html
│           └── index.html
├── generate.txt
├── output
│   ├── Final_bart.pdparams
│   │   ├── merges.txt
│   │   ├── model_config.json
│   │   ├── model_state.pdparams
│   │   ├── tokenizer_config.json
│   │   └── vocab.json
│   └── LCSTS.pdparams
│       ├── merges.txt
│       ├── model_config.json
│       ├── model_state.pdparams
│       ├── tokenizer_config.json
│       └── vocab.json
├── requirement.txt
├── task
│   ├── predict.py
│   ├── test.py
│   ├── train.py
│   └── unit_test.py
└── util
    ├── __init__.py
    ├── common_utils.py
    ├── csl_title_public.py
    ├── test_utils.py
    └── train_utils.py

Download Weight

Weight File: Final_bart.pdparams LCSTS.pdparams

Installation 🍻

pip install -r requirement.txt

Deploy(Show on Web Site) 🚀

python deploy/deploy.py

Running on http://0.0.0.0:5000/ (Press CTRL+C to quit)

Train

Note: Introducing wandb as training visualization tool.

Fine-tuning

wandb login # Wandb Token
python task/train.py

Test

python task/test.py

Predict

python task/predict.py