Downloads · 30 days
4
4% of all-time downloads
gpt2-imdb-finetuned
hinttt/gpt2-imdb-finetuned
gpt2-imdb-finetuned is a text generation model from hinttt. Use it when you need the model to write or continue text. It is set up for keras-hub. The card lists the license as mit.
- Training Loss: 1.0200 (after 1 epoch) - Hardware: Google Colab GPU/TPU
All-time downloads
90
Public
Repo size
996 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.h5498 MB · 100%
At a glance
- Task
- Text Generation
- Library
- keras-hub
- License
- mit
- Access
- Public
- Created
- Jun 3, 2026
- Updated
- Jun 3, 2026
- SHA
- 2da6d27d
Try a prompt
Model card
From the Hugging Face model README
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
import keras_hub
model = keras_hub.models.CausalLM.from_preset("hf://hinttt/gpt2-imdb-finetuned")
output = model.generate("This movie was", max_length=100)
- Task
- Text Generation
- Library
- keras-hub
- License
- mit
- Languages
- en
- Created
- Jun 3, 2026
- Updated
- Jun 3, 2026