shamikbose89 commited on
Commit
13ece46
·
1 Parent(s): 7acb61a

Update app.py

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