GPT2-IMDB is trained on IMDB dataset to generate a review for the given movie name(generates positive and negative reviews).
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="pt-sk/gpt2-imdb")
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("pt-sk/gpt2-imdb")
model = AutoModelForCausalLM.from_pretrained("pt-sk/gpt2-imdb")
- Downloads last month
- 10