File size: 2,313 Bytes
85488a7 e50223c 85488a7 89cc482 85488a7 f2a6fdd 85488a7 35674d5 85488a7 d868a86 85488a7 89cc482 4e3bbdf 89cc482 35674d5 85488a7 89cc482 85488a7 89cc482 35674d5 85488a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
---
language:
- vi
- en
tags:
- translation
widget:
- text: "vi: Bạn có thể giảm giá không?"
license: openrail
---
# ViT5 Translation
```python
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
model_name = "maihuyhoat/Translate"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
model.cuda()
inputs = [
"en: Too expensive.",
"en: Can you give me a discount?",
"en: Do you want to sell?",
"en: I want to buy.",
"en: One size larger.",
"en: One size smaller.",
"en: What do you do?",
"vi: Quá đắt.",
"vi: Bạn có thể giảm giá?",
"vi: Bạn có muốn bán không?",
"vi: Tôi muốn mua.",
"vi: Một kích thước lớn hơn.",
"vi: Một size nhỏ hơn.",
"vi: Bạn làm nghề gì?"
]
outputs = model.generate(tokenizer(inputs, return_tensors="pt", padding=True).input_ids.to('cuda'), max_length=50)
print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
# Expected outputs:
# ['vi: Quá đắt.',
# 'vi: Bạn có thể giảm giá không?',
# 'vi: Bạn có muốn bán không?',
# 'vi: Tôi muốn mua.',
# 'vi: Một kích thước lớn hơn.',
# 'vi: Một size nhỏ hơn.',
# 'vi: Bạn làm nghề gì?',
# 'en: Too expensive.',
# 'en: Can you reduce the price?',
# 'en: Do you want to sell?',
# 'en: I want to buy.',
# 'en: One size larger.',
# 'en: One size smaller.',
# 'en: What do you do?']
# Expected outputs:
# ['en: KHTN-CN ThuDo Academy offers courses in machine learning and artificial intelligence with the goal of enhancing Vietnam's competitiveness on the international stage.',
# 'en: Technologies such as deep learning and natural language processing are being widely applied in various fields such as healthcare, finance, and education.',
# 'vi: Trí tuệ nhân tạo đang thay đổi thế giới bằng cách cho phép máy móc học từ dữ liệu và thực hiện các nhiệm vụ thường yêu cầu trí tuệ con người.',
# 'vi: Sự hợp tác giữa ngành công nghiệp và học viện là rất quan trọng đối với sự tiến bộ của công nghệ AI và các ứng dụng của nó.']
```
|