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()