distlgpt2 / app.py
hf-dongpyo's picture
Update app.py
aa05842
raw
history blame contribute delete
493 Bytes
from transformers import pipeline, set_seed
import gradio as grad
gpt2_pipe = pipeline('text-generation', model = 'distilgpt2')
set_seed(42)
def generate(starting_text):
response = gpt2_pipe(starting_text, max_length = 20, num_return_sequences = 5)
return response
txt = grad.Textbox(lines = 1, label = 'English', placeholder = 'English Text Here')
out = grad.Textbox(lines = 1, label = 'Generated Text')
grad.Interface(
generate,
inputs = txt,
outputs = out
).launch()