Spaces:
Running
Running
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() |