Update app.py

app.py

@@ -9,6 +9,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 def download_argos_model(from_code, to_code):
     import argostranslate.package
+    print('Downloading model', from_code, to_code) 
     # Download and install Argos Translate package
     argostranslate.package.update_package_index()
     available_packages = argostranslate.package.get_available_packages()
@@ -26,8 +27,8 @@ input_text = st.text_input("Enter text to translate:")
 # Language options and mappings
 options = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
 langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es", "Italian": "it"}
-models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "Unbabel/Tower-Plus-2B", 
-          "Unbabel/TowerInstruct-Mistral-7B-v0.2", "Google", "Argos"]
+models = ["Helsinki-NLP", "Argos", "t5-base", "t5-small", "t5-large", "Unbabel/Tower-Plus-2B", 
+          "Unbabel/TowerInstruct-Mistral-7B-v0.2", "Google"]
 
 # Initialize session state if not already set
 if "sselected_language" not in st.session_state:
@@ -35,7 +36,7 @@ if "sselected_language" not in st.session_state:
 if "tselected_language" not in st.session_state:
     st.session_state["tselected_language"] = options[1]
 if "model_name" not in st.session_state:
-    st.session_state["model_name"] = models[0]
+    st.session_state["model_name"] = models[1]
 
 # Model selection FIRST
 model_name = st.selectbox("Select a model:", models, 
@@ -62,6 +63,8 @@ sl = langs[st.session_state["sselected_language"]]
 tl = langs[st.session_state["tselected_language"]]
 
 # Store selections
+st.session_state["sselected_language"] = sselected_language
+st.session_state["tselected_language"] = tselected_language
 st.session_state["model_name"] = model_name
 
 if model_name == 'Helsinki-NLP':
@@ -77,7 +80,6 @@ if model_name.startswith('t5'):
     tokenizer = T5Tokenizer.from_pretrained(model_name)
     model = T5ForConditionalGeneration.from_pretrained(model_name).to(device)
 
-
 st.write("Selected language combination:", sselected_language, " - ", tselected_language, "Selected model:", model_name)
 submit_button = st.button("Translate")
 translated_textarea = st.text("")