jharrison27's picture
Create app.py
094ae5a
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()
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 )
demo.launch()