File size: 804 Bytes
e1a9121 852fe1e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# -*- coding: utf-8 -*-
"""app.ipynb
Automatically generated by Colab.
Original file is located at
https://colab.research.google.com/drive/1-t9RYzLIxj4UDdky3Q_wyg29YwrPfJZN
"""
import gradio as gr
from transformers import pipeline
generator = pipeline("text-generation", model="gpt2")
def generate_text(prompt):
output = generator(prompt, max_new_tokens=50, do_sample=True, return_full_text=False)
return output[0]["generated_text"]
demo = gr.Interface(
fn=generate_text,
inputs=gr.Textbox(lines=4, placeholder="Type a short prompt here..."),
outputs=gr.Textbox(lines=4, label="Output"),
allow_flagging="never",
title="Text Generator Demo",
description="A simple Gradio app using GPT-2 from Hugging Face Transformers.",
theme="soft",
)
demo.queue().launch() |