Zeimoto commited on
Commit
9983262
·
1 Parent(s): 1f8b546

hardcoded 1000 max length

Browse files
Files changed (1) hide show
  1. translation.py +1 -1
translation.py CHANGED
@@ -16,7 +16,7 @@ def translate(text_to_translate: str) -> str:
16
  text_size = len(text_to_translate)*2
17
  tokenizer = AutoTokenizer.from_pretrained("unicamp-dl/translation-pt-en-t5")
18
  model = AutoModelForSeq2SeqLM.from_pretrained("unicamp-dl/translation-pt-en-t5")
19
- pten_pipeline = pipeline('text2text-generation', kwargs={"max_new_tokens": text_size},model=model, tokenizer=tokenizer)
20
  translated_text = pten_pipeline(text_to_translate)
21
 
22
  audit_elapsedtime(function="Finished translation", start=start)
 
16
  text_size = len(text_to_translate)*2
17
  tokenizer = AutoTokenizer.from_pretrained("unicamp-dl/translation-pt-en-t5")
18
  model = AutoModelForSeq2SeqLM.from_pretrained("unicamp-dl/translation-pt-en-t5")
19
+ pten_pipeline = pipeline('text2text-generation', kwargs={"max_new_tokens": "1000"}, model=model, tokenizer=tokenizer)
20
  translated_text = pten_pipeline(text_to_translate)
21
 
22
  audit_elapsedtime(function="Finished translation", start=start)