PereLluis13 commited on
Commit
8ef36bc
1 Parent(s): 5df3a7a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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": "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,
 
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,