Commit
•
37f3e35
1
Parent(s):
8ef36bc
Update README.md
Browse files
README.md
CHANGED
@@ -146,7 +146,7 @@ def extract_triplets_typed(text):
|
|
146 |
return triplets
|
147 |
|
148 |
# Load model and tokenizer
|
149 |
-
tokenizer = AutoTokenizer.from_pretrained("Babelscape/mrebel-large", src_lang="en_XX",
|
150 |
model = AutoModelForSeq2SeqLM.from_pretrained("Babelscape/mrebel-large")
|
151 |
gen_kwargs = {
|
152 |
"max_length": 256,
|
|
|
146 |
return triplets
|
147 |
|
148 |
# Load model and tokenizer
|
149 |
+
tokenizer = AutoTokenizer.from_pretrained("Babelscape/mrebel-large", src_lang="en_XX", tgt_lang="tp_XX") # Here we set English as source language. To change the source language just change it here or swap the first token of the input for your desired language
|
150 |
model = AutoModelForSeq2SeqLM.from_pretrained("Babelscape/mrebel-large")
|
151 |
gen_kwargs = {
|
152 |
"max_length": 256,
|