text-generation / app.py
Jezia's picture
Update app.py
88fd916
raw
history blame
517 Bytes
import gradio as gr
from gradio import inputs
description = "Text generation with a miniature GPT"
from huggingface_hub import from_pretrained_keras
model = from_pretrained_keras("huggingface/keras-io/text-generation-miniature-gpt")
interface = gr.Interface.load("huggingface/pranavpsv/gpt2-genre-story-generator",
title = "Text Generation with GPT-2",
inputs = [
gr.inputs.Textbox(lines=5, label="Text"),
],
description=description,
examples=[["A kid is playing with "]]
)
interface.launch()