gpt2_review_star3 / README.md
ehdwns1516's picture
Update README.md
9bdae66

gpt2_review_star3

  • This model has been trained as a review_body dataset with a star of 3 in the amazon_review 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: gpt2

Language: English

Training data: review_body dataset with a star of 3 in the amazon_review dataset.

Code: See Ainize Workspace

Usage

In Transformers

from transformers import AutoTokenizer, AutoModelWithLMHead
  
tokenizer = AutoTokenizer.from_pretrained("ehdwns1516/gpt2_review_star3")

model = AutoModelWithLMHead.from_pretrained("ehdwns1516/gpt2_review_star3")

generator = pipeline(
    "text-generation",
    model="ehdwns1516/gpt2_review_star3",
    tokenizer=tokenizer
)

context = "your context"

result = dict()
result[0] = generator(context)[0]