storresbusquets commited on
Commit
0c7f95c
1 Parent(s): 654d03b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ class GradioInference:
46
  self.classifier = pipeline("text-classification", model="lxyuan/distilbert-base-multilingual-cased-sentiments-student", return_all_scores=False)
47
 
48
  # Initialize Multilingual summary model
49
- self.tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/mT5_multilingual_XLSum")
50
  self.model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/mT5_multilingual_XLSum")
51
 
52
  self.config = {'max_new_tokens': 128, 'repetition_penalty': 1.1, 'temperature':0}
 
46
  self.classifier = pipeline("text-classification", model="lxyuan/distilbert-base-multilingual-cased-sentiments-student", return_all_scores=False)
47
 
48
  # Initialize Multilingual summary model
49
+ self.tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/mT5_multilingual_XLSum", truncation=True)
50
  self.model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/mT5_multilingual_XLSum")
51
 
52
  self.config = {'max_new_tokens': 128, 'repetition_penalty': 1.1, 'temperature':0}