Gpt2 Emotion

🚀 GPT - 2情感文本生成模型

本项目基于GPT - 2模型，可根据指定的情感标签生成相关的英文文本，为情感文本生成任务提供了便捷的解决方案。

🚀 快速开始

你可以使用以下代码快速体验模型的文本生成功能：

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

✨ 主要特性

• 情感驱动：支持基于不同情感标签（悲伤、喜悦、爱、愤怒、恐惧、惊讶）生成文本。
• 易于使用：借助transformers库的pipeline函数，简单几行代码即可实现文本生成。

📦 安装指南

本项目依赖于transformers库，你可以使用以下命令进行安装：

pip install transformers

💻 使用示例

基础用法

from transformers import pipeline

# 加载文本生成管道
generator = pipeline('text-generation', 'heegyu/gpt2-emotion')

# 定义输入提示
prompt = "sadness I'm so "
# 生成文本
print(generator(prompt)[0]['generated_text'])

📚 详细文档

输入提示要求

输入提示的起始词必须是以下之一：["sadness", "joy", "love", "anger", "fear", "surprise"]，以指定生成文本的情感倾向。

🔧 技术细节

本项目使用的模型为GPT - 2，它是一个基于Transformer架构的大型语言模型。通过在emotion数据集上进行训练，使得模型能够根据不同的情感标签生成相关的文本。

📄 许可证

本项目采用MIT许可证。