This repository contains the code for the following paper:
Khalil Mrini, Harpreet Singh, Franck Dernoncourt, Seunghyun Yoon, Trung Bui, Walter Chang, Emilia Farcas, Ndapa Nakashole
COLING 2022
Gyeongju, Republic of Korea, and Online
Use the following commands to install the requirements for this repository:
pip install -r requirements.txt
The main.py
file contains all the training and testing modes.
For example, use python3 main.py train_meqsum
to train a model for the MeQSum dataset.
You may train a summarizer for the GAR baseline by running the train_summarizer_for_gar.py
file.
Run the app_chatbot.py
file to start the Chatbot and evaluate answers. The DPR and GAR take about 40 to 50 seconds to retrieve an answer, but our model should be near immediate to retrieve one.