Spaces:
Runtime error
Runtime error
Commit
·
581be22
1
Parent(s):
2d0308a
Added
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import torch
|
|
3 |
from transformers import BertTokenizerFast, EncoderDecoderModel
|
4 |
|
5 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
6 |
-
ckpt =
|
7 |
tokenizer = BertTokenizerFast.from_pretrained(ckpt)
|
8 |
model = EncoderDecoderModel.from_pretrained(ckpt).to(device)
|
9 |
|
|
|
3 |
from transformers import BertTokenizerFast, EncoderDecoderModel
|
4 |
|
5 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
6 |
+
ckpt = "csebuetnlp/mT5_multilingual_XLSum"
|
7 |
tokenizer = BertTokenizerFast.from_pretrained(ckpt)
|
8 |
model = EncoderDecoderModel.from_pretrained(ckpt).to(device)
|
9 |
|