初始化项目,由ModelHub XC社区提供模型

Model: heegyu/gpt2-emotion
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-10-09 18:07:18 +08:00
commit 6706729f4a
12 changed files with 150496 additions and 0 deletions

25
README.md Normal file
View 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
```