TiberiuCristianLeon commited on
Commit
0d15afd
·
verified ·
1 Parent(s): 1c274df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,8 +14,8 @@ sselected_language = st.selectbox("Select a source language:", options)
14
  tselected_language = st.selectbox("Select a target language:", options)
15
  model_name = st.selectbox("Select a model:", models)
16
  # Display the selected language
17
- st.session_state["slanguage_selector"] = sselected_language
18
- st.session_state["tlanguage_selector"] = tselected_language
19
  st.session_state["model_name"] = model_name
20
  tokenizer = T5Tokenizer.from_pretrained(model_name)
21
  model = T5ForConditionalGeneration.from_pretrained(model_name)
@@ -31,4 +31,4 @@ if submit_button:
31
  # Decode the translated text
32
  translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
33
  # Display the translated text
34
- st.write(f"Translated Text from {sselected_language}:", translated_text)
 
14
  tselected_language = st.selectbox("Select a target language:", options)
15
  model_name = st.selectbox("Select a model:", models)
16
  # Display the selected language
17
+ st.session_state["sselected_language"] = sselected_language
18
+ st.session_state["tselected_language"] = tselected_language
19
  st.session_state["model_name"] = model_name
20
  tokenizer = T5Tokenizer.from_pretrained(model_name)
21
  model = T5ForConditionalGeneration.from_pretrained(model_name)
 
31
  # Decode the translated text
32
  translated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
33
  # Display the translated text
34
+ st.write(f"Translated text from {sselected_language} to {tselected_language}:", translated_text)