nb-nn-translation / README.md
pere's picture
Update README.md
67b22d8
|
raw
history blame
917 Bytes
---
language: no
license: CC-BY 4.0
tags:
- translation
datasets:
- oscar
widget:
- text: "Dette er en test!"
---
# Norwegian mT5 - Translation Bokmål Nynorsk
## Description
This is a sample reference model.
Here is an example of how to use the model from Python
```python
from transformers import T5ForConditionalGeneration, AutoTokenizer
model = T5ForConditionalGeneration.from_pretrained('.',from_flax=True)
tokenizer = AutoTokenizer.from_pretrained(".") # Or tokenizer = AutoTokenizer.from_pretrained("google/mt5-base")
inputs = tokenizer.encode("Det skal ifølge avisene være snakk om en femårskontrakt. Når den kontrakten utløper vil Messi være 39 år gammel, men de spanske avisene skriver at det ikke er ventet at han kommer til å fullføre kontrakten.", return_tensors="pt")
outputs = model.generate(inputs, max_length=255, num_beams=4, early_stopping=True)
print(tokenizer.decode(outputs[0]))
```