codeviking / lm-explorer-1

interactive explorer for language models

lm-explorer

interactive explorer for language models (currently only OpenAI GPT-2)

Running with Docker

# This creates a local directory where the model can be cached so you don't
# have to download it everytime you execute 'docker run'.
$ mkdir -p /$HOME/.pytorch_pretrained_bert
$ docker build -t lm-explorer:latest .
$ docker run -p 8000:8000 \
    -v /$HOME/.pytorch_pretrained_bert:/root/.pytorch_pretrained_bert \
    -v $(pwd):/local \
    lm-explorer:latest \
    python app.py --port 8000 --dev

Running without Docker

First create and activate a Python 3.6 (or later) virtual environment. Then install the requirements

$ pip install -r requirements.txt

and start the app

$ python app.py --port 8000 --dev