text-generator / app.py
mimic12's picture
Add application file
90879b3
raw
history blame
385 Bytes
import gradio as gr
from transformers import pipeline
generator = pipeline("text-generation", model="gpt2")
def generate_text(prompt):
return generator(prompt, max_new_tokens=50)[0]["generated_text"]
gr.Interface(fn=generate_text,
inputs=gr.Textbox(label="프롬프트 입력"),
outputs="text",
title="GPT-2 텍스트 생성기").launch()