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