agaliano commited on
Commit
e54dff5
verified
1 Parent(s): 4fd5dec

Add Usage section

Browse files
Files changed (1) hide show
  1. README.md +21 -3
README.md CHANGED
@@ -1,3 +1,21 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+
5
+ ## Usage
6
+
7
+ ```python
8
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
9
+
10
+ model = AutoModelForSeq2SeqLM.from_pretrained("Transducens/IbRo-nllb")
11
+ tokenizer = AutoTokenizer.from_pretrained("Transducens/IbRo-nllb")
12
+
13
+ tokenizer.src_lang = "spa_Latn"
14
+
15
+ sentence = "芦Actualmente, tenemos ratones de cuatro meses de edad que antes sol铆an ser diab茅ticos y que ya no lo son禄, agreg贸."
16
+ inputs = tokenizer(sentence, return_tensors="pt")
17
+
18
+ translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id["arg_Latn"])
19
+
20
+ print(tokenizer.batch_decode(translated_tokens, skip_special_tokens=True))
21
+ ```