Edit model card
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

ehdwns1516/gpt3-kor-based_gpt2_review_SR5

  • This model has been trained Korean dataset as a star of 5 in the naver shopping reivew dataset.
  • Input text what you want to generate review.
  • If the context is longer than 1200 characters, the context may be cut in the middle and the result may not come out well.

review generator DEMO: Ainize DEMO

review generator API: Ainize API

Model links for each 1 to 5 star

Overview

Language model: gpt3-kor-small_based_on_gpt2

Language: Korean

Training data: review_body dataset with a star of 5 in the naver shopping reivew dataset.

Code: See Ainize Workspace

Usage

In Transformers

from transformers import AutoTokenizer, AutoModelWithLMHead
  
tokenizer = AutoTokenizer.from_pretrained("ehdwns1516/gpt3-kor-based_gpt2_review_SR5")

model = AutoModelWithLMHead.from_pretrained("ehdwns1516/gpt3-kor-based_gpt2_review_SR5")

generator = pipeline(
    "text-generation",
    model="ehdwns1516/gpt3-kor-based_gpt2_review_SR5",
    tokenizer=tokenizer
)

context = "your context"

result = dict()
result[0] = generator(context)[0]
Downloads last month
38