nb-nn-translation / README.md
pere's picture
Update README.md
67b22d8
|
raw
history blame
917 Bytes
metadata
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

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]))