shamikbose89 commited on
Commit
ad7f067
·
1 Parent(s): acf6536

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
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)