TiberiuCristianLeon commited on
Commit
61e85d4
·
verified ·
1 Parent(s): ee74465

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,11 +12,11 @@ models = ["t5-base", "t5-small", "opus-mt-en-ro"]
12
  # Create the select box
13
  sselected_language = st.selectbox("Select a source language:", options)
14
  tselected_language = st.selectbox("Select a target language:", options)
15
- model_selector = 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_selector
20
  tokenizer = T5Tokenizer.from_pretrained(model_name)
21
  model = T5ForConditionalGeneration.from_pretrained(model_name)
22
  st.write("Selected language combination:", selected_language, ' - ', selected_language)
 
12
  # Create the select box
13
  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)
22
  st.write("Selected language combination:", selected_language, ' - ', selected_language)