text-generation / app.py
Jezia's picture
Update app.py
d810c79
raw
history blame
472 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(model,
title = "Text Generation with GPT-2",
inputs = [
gr.inputs.Textbox(lines=5, label="Text"),
],
description=description,
examples=[["A kid is playing with "]]
)
interface.launch()