Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
|
|
28 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
|
29 |
translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='ron_Latn', max_length<=360)
|
30 |
translated_text = translator(input_text)
|
31 |
-
return translated_text[0][
|
32 |
else:
|
33 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
34 |
model = T5ForConditionalGeneration.from_pretrained(model_name, device_map="auto")
|
|
|
28 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name_full)
|
29 |
translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang='eng_Latn', tgt_lang='ron_Latn', max_length<=360)
|
30 |
translated_text = translator(input_text)
|
31 |
+
return translated_text[0]['translation_text']
|
32 |
else:
|
33 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
34 |
model = T5ForConditionalGeneration.from_pretrained(model_name, device_map="auto")
|