File size: 473 Bytes
730dc34
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from transformers import pipeline, set_seed
import gradio as gr

model_name = "distilgpt2"
gpt2_pipeline = pipeline("text-generation", model=model_name)
set_seed(42)

def generate(text):
    response = gpt2_pipeline(text, max_length=20, num_return_sequences=5)
    return response

in_text = gr.Textbox(lines=1, label="English", placeholder="English text here")
out = gr.Textbox(lines=1, label="Generated text")

gr.Interface(generate, inputs=in_text, outputs=out).launch()