This repository uses HuggingFace's GPT2 Implementation and exposes an creates a nice user interface for testing GPT2 power. Even though it may not be exactly as good as authors' original tensorflow implementation, it still surprises sometimes.
New: Added GPT-2 Medium model support (large model requires >2GB RAM which is not supported by Cloud Run)
To get started on this project, follow these steps:
- Clone:
git clone https://github.com/NaxAlpha/gpt-2xy.gitandcd gpt-2xy - Install Dependencies:
pip install -r requirements.txt - Run server:
python main.py - Open Web Browser and visit:
http://localhost:8080/ - Type
Microsoft and Googleand pressGENERATE TEXT - Enjoy!
You may also want to build and run this project in docker container
- Build:
docker build -t gpt-2xy .(will take some time) - Run:
docker run -p 8080:8080 --rm -d gpt-2xy
If successful, you will see something like this:
This repository contains code for my medium tutorial on Deploy Machine Learning Model in Google Cloud using Cloud Run