Spaces:

TiberiuCristianLeon
/

GradioTranslate

Running

TiberiuCristianLeon commited on Jan 6

Commit

e3f013c

verified ·

1 Parent(s): 4fd4915

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=langs[sselected_language][1], tgt_lang=langs[tselected_language][1])
-        translated_text = translator(input_text)
         return translated_text[0]['translation_text']
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)

         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=langs[sselected_language][1], tgt_lang=langs[tselected_language][1])
+        translated_text = translator(input_text, max_length=512)
         return translated_text[0]['translation_text']
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)