Spaces:
Runtime error
Runtime error
Commit
·
ad7f067
1
Parent(s):
acf6536
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
from gradio.mix import Parallel
|
3 |
+
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
|
4 |
+
|
5 |
+
distilbart_model = "https://huggingface.co/sshleifer/distilbart-cnn-12-6"
|
6 |
+
model_name = "shamikbose89/mt5-small-finetuned-arxiv-cs-full"
|
7 |
+
|
8 |
+
base_model = gr.Interface.load(distilbart_model, title="DistilBART output")
|
9 |
+
my_model = gr.Interface.load(model_name, title = "Finetuned model output", inputs = text, outputs=text)
|
10 |
+
Parallel(base_model, my_model)
|