Tapanat commited on
Commit
9a38909
·
1 Parent(s): eb58d9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import streamlit as st
8
  from transformers import BartForConditionalGeneration, BartTokenizer
9
 
10
  # Load the pre-trained BART model and tokenizer
11
- model_name = "facebook/bart-large-cnn"
12
  model = BartForConditionalGeneration.from_pretrained(model_name)
13
  tokenizer = BartTokenizer.from_pretrained(model_name)
14
 
 
8
  from transformers import BartForConditionalGeneration, BartTokenizer
9
 
10
  # Load the pre-trained BART model and tokenizer
11
+ model_name = "csebuetnlp/mT5_multilingual_XLSum"
12
  model = BartForConditionalGeneration.from_pretrained(model_name)
13
  tokenizer = BartTokenizer.from_pretrained(model_name)
14