Spaces:
Runtime error
Runtime error
Zeimoto
commited on
Commit
·
9983262
1
Parent(s):
1f8b546
hardcoded 1000 max length
Browse files- 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',
|
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)
|