Spaces:
Sleeping
Sleeping
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() | |