File size: 365 Bytes
fec632f
 
 
 
2af52d7
e765751
2af52d7
fec632f
 
2af52d7
 
1
2
3
4
5
6
7
8
9
10
11
from transformers import pipeline, set_seed
from random import seed, choice
generator = pipeline('text-generation', model='gpt2')
set_seed(seed())
import gradio as gr

def gpt2(string):
    text = generator(text, max_length=250, num_return_sequences=1)
    return text[0]['generated_text']
iface = gr.Interface(fn=gpt2, inputs="text", outputs="text")
iface.launch()