next word prediction

Próxima previsão de palavras

Transformer 2 generativo pré-terenciado (GPT-2) para modelagem de idiomas usando a biblioteca Pytorch-Transformers.

Instalação

Requer python> = 3,5, pytorch> = 1.6.0, pytorch-transformers> = 1.2.0

pip install next-word-prediction

Como usar

 > >> from next_word_prediction import GPT2
> >> gpt2 = GPT2 ()
> >> text = "The course starts next"
> >> gpt2 . predict_next ( text , 5 )
The course starts next [ 'week' , 'to' , 'month' , 'year' , 'Monday' ]

Demonstração via streamlit

streamlit run app/run.py