Spaces:

TiberiuCristianLeon
/

GradioTranslate

Running

TiberiuCristianLeon commited on Jan 5

Commit

7810728

verified ·

1 Parent(s): fe93b05

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
         tokenizer = AutoTokenizer.from_pretrained(model_name_full)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='ron_Latn')
-        translated_text = translator(text, max_length<=360)
         return translated_text
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)

         tokenizer = AutoTokenizer.from_pretrained(model_name_full)
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='ron_Latn')
+        translated_text = translator(input_text, max_length<=360)
         return translated_text
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)