|
# gpt2_review_star4 |
|
|
|
* This model has been trained as a review_body dataset with a star of 4 in the [amazon_review dataset](https://huggingface.co/datasets/amazon_reviews_multi). |
|
* 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](https://main-review-generator-ehdwns1516.endpoint.ainize.ai/) |
|
|
|
review generator API: [Ainize API](https://ainize.web.app/redirect?git_repo=https://github.com/ehdwns1516/review_generator) |
|
|
|
## Model links for each 1 to 5 star |
|
* [ehdwns1516/gpt2_review_star1](https://huggingface.co/ehdwns1516/gpt2_review_star1) |
|
* [ehdwns1516/gpt2_review_star2](https://huggingface.co/ehdwns1516/gpt2_review_star2) |
|
* [ehdwns1516/gpt2_review_star3](https://huggingface.co/ehdwns1516/gpt2_review_star3) |
|
* [ehdwns1516/gpt2_review_star4](https://huggingface.co/ehdwns1516/gpt2_review_star4) |
|
* [ehdwns1516/gpt2_review_star5](https://huggingface.co/ehdwns1516/gpt2_review_star5) |
|
|
|
## Overview |
|
|
|
Language model: [gpt2](https://huggingface.co/gpt2) |
|
|
|
Language: English |
|
|
|
Training data: review_body dataset with a star of 4 in the [amazon_review dataset](https://huggingface.co/datasets/amazon_reviews_multi). |
|
|
|
Code: See [Ainize Workspace](https://ainize.ai/workspace/create?imageId=hnj95592adzr02xPTqss&git=https://github.com/ehdwns1516/gpt2_review_fine-tunning_note) |
|
|
|
## 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_star4", |
|
tokenizer=tokenizer |
|
) |
|
|
|
context = "your context" |
|
|
|
result = dict() |
|
result[0] = generator(context)[0] |
|
``` |
|
|