Update README.md
Browse files
README.md
CHANGED
@@ -26,8 +26,8 @@ else:
|
|
26 |
print('No GPU available, using the CPU instead.')
|
27 |
device = torch.device("cpu")
|
28 |
|
29 |
-
model = T5ForConditionalGeneration.from_pretrained("
|
30 |
-
tokenizer = T5Tokenizer.from_pretrained("
|
31 |
model.to(device)
|
32 |
|
33 |
src = "Indonesia phỏng đoán nguyên nhân tàu ngầm chở 53 người mất tích bí ẩn"
|
|
|
26 |
print('No GPU available, using the CPU instead.')
|
27 |
device = torch.device("cpu")
|
28 |
|
29 |
+
model = T5ForConditionalGeneration.from_pretrained("NlpHUST/t5-vi-en-small")
|
30 |
+
tokenizer = T5Tokenizer.from_pretrained("NlpHUST/t5-vi-en-small")
|
31 |
model.to(device)
|
32 |
|
33 |
src = "Indonesia phỏng đoán nguyên nhân tàu ngầm chở 53 người mất tích bí ẩn"
|