text_generator / app.py
Wootang01's picture
Update app.py
3dc4cab
raw
history blame
683 Bytes
#import libraries and dependencies
import gradio as gr
from gradio.mix import Parallel, Series
#instantiate variables as functions
model1 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
model2 = gr.Interface.load("huggingface/google/flan-t5-xl")
#model3 = gr.Interface.load("huggingface/Wootang01/distilgpt2-finetuned-prayerjournals")
model4 = gr.Interface.load("huggingface/bigscience/bloomz-1b7")
#translator = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-zh-en")
output = gr.Parallel(model4, model2, model1).launch()
#demo = gr.Series(translator, output, inputs=gr.Textbox(lines=5, label="Input Text"), title=title, description=description)
demo.launch(debug=True)