File size: 291 Bytes
7681358
4c476ba
7681358
a223eb7
3e5ffb2
ab81ec8
a223eb7
 
ab81ec8
b5485dd
1
2
3
4
5
6
7
8
9
10
import gradio as gr
from transformers import pipeline

generator = pipeline('text-generation', model='gpt2')

def generate(text):
    result = generator(text)
    return result[0]['generated_text']

gr.Interface(fn=generate, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Textbox()).launch()