Create README.md

README.md

@@ -0,0 +1,25 @@
+---
+license: mit
+datasets:
+- emotion
+language:
+- en
+library_name: transformers
+---
+
+- **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
+```