Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import transformers as tr | |
| import numpy as np | |
| generator1 = gr.Interface.load("huggingface/gpt2-large") | |
| generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B") | |
| generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B") | |
| demo = gr.Blocks() | |
| def f1(x): | |
| return generator1(x) | |
| def f2(x): | |
| return generator2(x) | |
| def f3(x): | |
| return generator3(x) | |
| with demo: | |
| textIn = gr.Textbox() | |
| textOut1 = gr.Textbox() | |
| bt1 = gr.Button("Re-run") | |
| textOut2 = gr.Textbox() | |
| textOut3 = gr.Textbox() | |
| b1 = gr.Button("gpt2-large") | |
| b2 = gr.Button("gpt-neo-2.7B") | |
| b3 = gr.Button("gpt-j-6B") | |
| b1.click(f1, inputs=textIn, outputs=textOut1 ) | |
| b2.click(f2, inputs=textIn, outputs=textOut2 ) | |
| b3.click(f3, inputs=textIn, outputs=textOut3 ) | |
| bt1.click(f3, inputs=textOut1, outputs=textOut2 ) | |
| demo.launch() | |