Update README.md
Browse filesFixing inference example
README.md
CHANGED
|
@@ -48,12 +48,15 @@ It is designed to handle the complexities of journalistic text, focusing on the
|
|
| 48 |
## 💻 Usage
|
| 49 |
|
| 50 |
```python
|
| 51 |
-
from transformers import
|
| 52 |
|
| 53 |
-
|
|
|
|
| 54 |
|
| 55 |
text = "ސީރިއާގެ ކަންކަމުގެ ތެރެއަށް އިސްރާއީލް ނުވަނުމަށް ޓްރަމްޕް ގޮވާލައްވައިފި"
|
| 56 |
-
|
| 57 |
|
| 58 |
-
|
|
|
|
|
|
|
| 59 |
# Output: Syria ge kankamuge thereah Israel nuvanumah Trump govaalavvaifi
|
|
|
|
| 48 |
## 💻 Usage
|
| 49 |
|
| 50 |
```python
|
| 51 |
+
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 52 |
|
| 53 |
+
tokenizer = AutoTokenizer.from_pretrained("Neobe/dhivehi-byt5-thaana2latin-v1")
|
| 54 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("Neobe/dhivehi-byt5-thaana2latin-v1")
|
| 55 |
|
| 56 |
text = "ސީރިއާގެ ކަންކަމުގެ ތެރެއަށް އިސްރާއީލް ނުވަނުމަށް ޓްރަމްޕް ގޮވާލައްވައިފި"
|
| 57 |
+
inputs = tokenizer(text, return_tensors="pt")
|
| 58 |
|
| 59 |
+
outputs = model.generate(**inputs, max_length=256)
|
| 60 |
+
|
| 61 |
+
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 62 |
# Output: Syria ge kankamuge thereah Israel nuvanumah Trump govaalavvaifi
|