Gpt2 Emotion

🚀 Heegyu氏のGPT - 2モデル

このプロジェクトは、Heegyu Kim氏によって開発されたGPT - 2ベースのテキスト生成モデルです。英語に対応し、MITライセンスの下で公開されています。主に感情に関連したテキスト生成に特化しています。

🚀 クイックスタート

このモデルを使用するには、transformersライブラリを利用します。以下のコードで簡単にテキスト生成を行うことができます。

💻 使用例

基本的な使用法

# コードとコメントはそのまま保持
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

📄 ライセンス

このモデルはMITライセンスの下で公開されています。