Downloads · 30 days
79
0% of all-time downloads
kogpt2-emotion-chatbot
withU/kogpt2-emotion-chatbot
kogpt2-emotion-chatbot is a text generation model from withU. Use it when you need the model to write or continue text. It is set up for transformers.
kogpt2 on hugging face Transformers for Psychological Counseling - full project link
All-time downloads
23.4K
Public
Repo size
1 GB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.bin510 MB · 99%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- May 12, 2022
- Updated
- May 16, 2022
- SHA
- 2dd5ffb2
Try a prompt
Spaces
Model card
From the Hugging Face model README
KoGPT2-emotion-chatbot
kogpt2 on hugging face Transformers for Psychological Counseling
how to use
from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
model = GPT2LMHeadModel.from_pretrained("withU/kogpt2-emotion-chatbot")
tokenizer = PreTrainedTokenizerFast.from_pretrained("withU/kogpt2-emotion-chatbot")
input_ids = tokenizer.encode("안녕", add_special_tokens=False, return_tensors="pt")
output_sequences = model.generate(input_ids=input_ids, do_sample=True, max_length=80, num_return_sequences=4)
for generated_sequence in output_sequences:
generated_sequence = generated_sequence.tolist()
print("GENERATED SEQUENCE : {0}".format(tokenizer.decode(generated_sequence, clean_up_tokenization_spaces=True)))
dataset finetuned on
references
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- Created
- May 12, 2022
- Updated
- May 16, 2022