Spaces:
Running
Running
import gradio as gr | |
from transformers import pipeline | |
english_translator = gr.load(name="spaces/gradio/english_translator") | |
english_generator = pipeline("text-generation", model="distilgpt2") | |
def generate_text(text): | |
english_text = english_generator(text)[0]["generated_text"] # type: ignore | |
german_text = english_translator(english_text) | |
return english_text, german_text | |
with gr.Blocks() as demo: | |
with gr.Row(): | |
with gr.Column(): | |
seed = gr.Text(label="Input Phrase") | |
with gr.Column(): | |
english = gr.Text(label="Generated English Text") | |
german = gr.Text(label="Generated German Text") | |
btn = gr.Button("Generate") | |
btn.click(generate_text, inputs=[seed], outputs=[english, german]) | |
gr.Examples(["My name is Clara and I am"], inputs=[seed]) | |
if __name__ == "__main__": | |
demo.launch() | |