Skip to content
This repository has been archived by the owner on Nov 27, 2022. It is now read-only.
/ doctor_chatbot Public archive

Flask doctor chatbot that produces a response for patient's query related to COVID-19. The app is designed based on GPT-2 pretrained on dialogpt weights.

Notifications You must be signed in to change notification settings

chophilip21/doctor_chatbot

Repository files navigation

covid_dialgue

  • This repository is created based on my SFU CMPT 825 Final Project, and the following is our report: https://drive.google.com/file/d/1D2BLTjKHPg_1zGe4kGyO7YPR8brx34Ed/view?usp=sharing
  • The implementation here uses GPT-2 structure pretrained with DialoGPT weights provided from Microsoft.
  • If you would like access to the custom dataset I have used for training, please email pycho@sfu.ca.
  • This dataset is custom made and it's very small. GPT-2 model needs a lot of data to function properly and NLP-aug library was used to increase the size of the dataset. Refer to dataset.py for more information.

Instrunctions for running the app

  • Install Virtual environemnt, and simply pip install requirements.txt
  • If you just want to test the app, just run python app.py
  • If you want to train from scratch, run train.py.

Demo

Below is the live demo of the app.py. The app uses Javascript for the front and simple Python Flask as backend. It was originally launched on Heroku sever, but later pulled down as the files are too large to meet Heroku limit. I plan to explore other options like AWS EC2 in the future. At the moment, it only runs on localhost server.

demo

About

Flask doctor chatbot that produces a response for patient's query related to COVID-19. The app is designed based on GPT-2 pretrained on dialogpt weights.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published