mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-27 02:30:17 +08:00
27 lines
860 B
Python
27 lines
860 B
Python
import gradio as gr
|
|
|
|
from transformers import pipeline
|
|
|
|
english_translator = gr.Blocks.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"]
|
|
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() |