PereLluis13
commited on
Commit
•
8ef36bc
1
Parent(s):
5df3a7a
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", "tgt_lang"
|
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,
|