pere commited on
Commit
67b22d8
1 Parent(s): d83f447

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -12,4 +12,15 @@ widget:
12
 
13
  ## Description
14
 
15
- This is a sample reference model.
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
  ## Description
14
 
15
+ This is a sample reference model.
16
+
17
+ Here is an example of how to use the model from Python
18
+ ```python
19
+ from transformers import T5ForConditionalGeneration, AutoTokenizer
20
+ model = T5ForConditionalGeneration.from_pretrained('.',from_flax=True)
21
+ tokenizer = AutoTokenizer.from_pretrained(".") # Or tokenizer = AutoTokenizer.from_pretrained("google/mt5-base")
22
+ 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")
23
+ outputs = model.generate(inputs, max_length=255, num_beams=4, early_stopping=True)
24
+ print(tokenizer.decode(outputs[0]))
25
+
26
+ ```