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