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