File size: 317 Bytes
0808a61 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# -*- coding: utf-8 -*-
from transformers import pipeline, set_seed
generator = pipeline('text-generation', model='gpt2')
import gradio as gr
def func(text):
generated = generator(text)
return generated
app = gr.Interface(fn=func, inputs="textbox", outputs="textbox", title="Text Generator")
app.launch()
|