TiberiuCristianLeon commited on
Commit
30c9436
·
verified ·
1 Parent(s): 1e9ddeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -13,7 +13,6 @@ models = ["t5-base", "t5-small", "opus-mt-en-ro"]
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["sselected_language"] = sselected_language
18
  st.session_state["tselected_language"] = tselected_language
19
  st.session_state["model_name"] = model_name
@@ -25,7 +24,7 @@ translated_textarea = st.text("")
25
 
26
  # Handle the submit button click
27
  if submit_button:
28
- prompt = f'translate {sselected_language} to {tselected_language}: {input_text}'
29
  print(prompt)
30
  input_ids = tokenizer.encode(prompt, return_tensors='pt')
31
  # Perform translation
 
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
  st.session_state["sselected_language"] = sselected_language
17
  st.session_state["tselected_language"] = tselected_language
18
  st.session_state["model_name"] = model_name
 
24
 
25
  # Handle the submit button click
26
  if submit_button:
27
+ prompt = f'Prompt: translate {sselected_language} to {tselected_language}: {input_text}'
28
  print(prompt)
29
  input_ids = tokenizer.encode(prompt, return_tensors='pt')
30
  # Perform translation