File size: 369 Bytes
ad7f067
 
 
13ece46
 
ad7f067
7acb61a
 
ad7f067
1
2
3
4
5
6
7
8
9
import gradio as gr
from gradio.mix import Parallel

distilbart_model = "huggingface/sshleifer/distilbart-cnn-12-6"
model_name = "huggingface/shamikbose89/mt5-small-finetuned-arxiv-cs-full"

base_model = gr.Interface.load(distilbart_model)
my_model = gr.Interface(model_name, title = "Finetuned model output", inputs = text, outputs=text)
Parallel(base_model, my_model)