shamikbose89's picture
Update app.py
61f7b2e
raw
history blame
361 Bytes
import gradio as gr
from gradio.mix import Parallel
def cr
distilbart_model = "huggingface/sshleifer/distilbart-cnn-12-6"
model_name = "huggingface.co/google/pegasus-large"
base_model = gr.Interface.load(distilbart_model)
my_model = gr.Interface.load(model_name, title = "Finetuned model output", inputs = "text", outputs="text")
Parallel(base_model, my_model)