Modelpublic

hinttt/gpt2-imdb-finetuned

sourceHugging Facemitupdated 4mo agoView on Hugging Face

0likes13downloads

Model Card

GPT-2 Movie Review Generator

Performance Metrics

—Training Loss: 1.0200 (after 1 epoch)
—Hardware: Google Colab GPU/TPU

Example Generation

Prompt: This movie was

Output:

one of the first I saw in theaters and I was so excited to see it again...

How to use

python

import keras_hub
model = keras_hub.models.CausalLM.from_preset("hf://hinttt/gpt2-imdb-finetuned")
output = model.generate("This movie was", max_length=100)