初始化项目,由ModelHub XC社区提供模型
Model: heegyu/gpt2-emotion Source: Original Platform
This commit is contained in:
25
README.md
Normal file
25
README.md
Normal file
@@ -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
|
||||
```
|
||||
Reference in New Issue
Block a user