ArissBandoss commited on
Commit
a892700
·
verified ·
1 Parent(s): e05b01e

Update goai_helpers/goai_traduction.py

Browse files
Files changed (1) hide show
  1. goai_helpers/goai_traduction.py +1 -0
goai_helpers/goai_traduction.py CHANGED
@@ -64,6 +64,7 @@ def translate_chunk(text, src_lang, tgt_lang):
64
  #model_id = "ArissBandoss/nllb-200-3.3B-mos-fr-bidirectional-peft"
65
 
66
  text = mark_numbers(text)
 
67
 
68
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
69
  model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)
 
64
  #model_id = "ArissBandoss/nllb-200-3.3B-mos-fr-bidirectional-peft"
65
 
66
  text = mark_numbers(text)
67
+ print(text)
68
 
69
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=auth_token)
70
  model = AutoModelForSeq2SeqLM.from_pretrained(model_id, token=auth_token).to(device)