GPU enabled environment is required.
numpy
PyTorch >= 1.0
pytorch_pretrained_bert
tqdm
redis
flask
User can get the prediction result by sending request to server.
python3 send_request [--title TITLE] [--tags TAGS [TAGS ...]] [--api_url API_URL]
- YenTing Lin - GitHub
This project is licensed under the MIT License - see the LICENSE.md file for details
- pytorch_pretrained_bert is cloned from https://github.com/huggingface/pytorch-pretrained-BERT
- Zero-shot classification for text is inspired by https://arxiv.org/abs/1712.05972