File size: 385 Bytes
90879b3
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
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()