TiberiuCristianLeon commited on
Commit
71f8c73
·
verified ·
1 Parent(s): bdb24ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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][translation_text]
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")