Skip to content

Small example of a web server with a text generator in multiple languages.

Notifications You must be signed in to change notification settings

cduguet/text-generator-demo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text generator demo

This is a smallm repository for a web app showcasing text-generators in multiple languages. The app is a starlette web-app and its based on the starter app recommended by fastai available to download.

The English model for prediction is an AWD-LSTM architecture using a pretrained model on WikiText-103.

The Spanish model is trained by myself on WikiText-103, and it's code is available here.

To-Do

A List of nice-to-have improvements I'd like to integrate in the future include but are not limited to:

  • Using prediction with beam search
  • Fine-Tune on a specific topic dataset (probably Fashion or Legal)
  • Train a BERT architecture
  • Train a German Network

About

Small example of a web server with a text generator in multiple languages.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages