Palashio / libra

Ergonomic machine learning for everyone.
http://libradocs.org/
MIT License
1.91k stars 109 forks source link

Text Generation Query Implemented #329

Closed umangj123 closed 4 years ago

umangj123 commented 4 years ago

name: b) NLP Text Generation about: Text generation with hugging face labels: feature

This pull request closes #257

- What I did I created functonality for text generation - How I did it using the gpt2 model from hugging face i was able to take in several parameters and then generate text based on the initial text they provide. In the end it should print out generated text as well as return self.models['Generated Text'] with the generated text inside - How to verify it Tests are included in tests.py

This pull request adds a new feature to libra. @Palashio, could you please take a look at it?