Spaces:
Runtime error
Runtime error
import gradio as gr | |
from gradio.mix import Parallel | |
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM | |
distilbart_model = "https://huggingface.co/sshleifer/distilbart-cnn-12-6" | |
model_name = "shamikbose89/mt5-small-finetuned-arxiv-cs-full" | |
base_model = gr.Interface.load(distilbart_model, title="DistilBART output") | |
my_model = gr.Interface.load(model_name, title = "Finetuned model output", inputs = text, outputs=text) | |
Parallel(base_model, my_model) |