You can deploy this model and start earning money today!

ehdwns1516/
gpt2_review_star5

Name: gpt2_review_star5
Availability: InStock

text-generation-inference

endpoints_compatible

Views

No views yet

gpt2_review_star5

This model has been trained as a review_body dataset with a star of 5 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 5 in the amazon_review dataset.

Code: See Ainize Workspace

Usage

In Transformers

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

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

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

context = "your context"

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