A PyTorch/Flask ChatBot for use in Fortify/Debricked demonstrations

Setup Python virtual environment:

$ python -m venv venv
$ .\venv\Scripts\Activate.ps1	[Windows]
$ . venv/bin/activate			[Linux/UNIX]

Install dependencies:

$ (venv) pip install -r requirements.txt

Install nltk package:

$ (venv) python
>>> import nltk
>>> nltk.download('punkt')

Train the model:

$ (venv) python train.py

This will dump data.pth file. And then run the following command to test it in the console.

$ (venv) python chat.py

Kevin Lee [email protected]