Modelpublic

heegyu/gpt2-emotion

sourceHugging Facemitupdated 4y agoView on Hugging Face

2likes117downloads

Model Card

—Developed by: Heegyu Kim
—Model type: GPT-2
—Language(s) (NLP): English
—License: MIT

Uses

from transformers import pipeline

generator = pipeline('text-generation', 'heegyu/gpt2-emotion')

prompt = "sadness I'm so " # start token should be one of ["sadness", "joy", "love", "anger", "fear", "surprise"]
print(generator(prompt)[0]['generated_text'])
>>> sadness I'm so tired of seeing all the stupid things that i ve learned from past years that i feel like ive been so stupid and blah and then i feel like ive just wasted all my energy doing stupid shit like how i never