shamikbose89 commited on
Commit
f2ef06d
1 Parent(s): f585723

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from gradio.mix import Parallel
3
 
4
  distilbart_model = "huggingface/sshleifer/distilbart-cnn-12-6"
5
- model_name = "huggingface/google/pegasus-large"
6
  base_model = gr.Interface.load(distilbart_model)
7
  my_model = gr.Interface.load(model_name, title = "Finetuned model output", inputs = "text", outputs="text")
8
  Parallel(base_model, my_model)
 
2
  from gradio.mix import Parallel
3
 
4
  distilbart_model = "huggingface/sshleifer/distilbart-cnn-12-6"
5
+ model_name = "huggingface/sshleifer/distill-pegasus-xsum-16-4"
6
  base_model = gr.Interface.load(distilbart_model)
7
  my_model = gr.Interface.load(model_name, title = "Finetuned model output", inputs = "text", outputs="text")
8
  Parallel(base_model, my_model)