Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 import os, gc
 import httpx
 
+
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Language options and mappings
 favourite_langs = {"Romanian": "ro", "German": "de", "English": "en", "-----": "-----"}
@@ -608,8 +609,8 @@ with tcol:
                                       index=langs.index(st.session_state["tselected_language"]))
 
 # Language codes
-sl = langs[st.session_state["sselected_language"]]
-tl = langs[st.session_state["tselected_language"]]
+sl = name_to_iso1[st.session_state["sselected_language"]]
+tl = name_to_iso1[st.session_state["tselected_language"]]
 
 # Store selections
 st.session_state["sselected_language"] = sselected_language