text_generator / app.py
Wootang01's picture
Change the gpt-j-6B language model to another model.
a0ec69f
raw
history blame
369 Bytes
import gradio as gr
from gradio.mix import Parallel
title="My First Text Generator"
description ="Input text"
model1=gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B")
model2=gr.Interface.load("huggingface/gpt2")
model3=gr.Interface.load("huggingface/EleutherAI/gpt-neo-125M")
gr.Parallel(model1, model2, model3, title=title, description =description).launch()