Spaces:
Runtime error
Runtime error
storresbusquets
commited on
Commit
•
0c7f95c
1
Parent(s):
654d03b
Update app.py
Browse files
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}
|