more mods

app.py

@@ -15,20 +15,20 @@ def load_model(model_name):
 
 model = load_model("facebook/nllb-200-distilled-600M")
 
-default_value = "UN Chief says there is no military solution in Syria"
-sent = st.text_area("Input Your Text Here", default_value, height = 275)
 src_lang_selection = st.radio(
     "Select Your Source Language:",
     ('English', 'Vietnamese'))
+if src_lang_selection == "English":
+    src_lang = "eng_Latn"
+    tgt_lang = "vie_Latn"
+else:
+    src_lang = "vie_Latn"
+    tgt_lang = "eng_Latn"
+default_value = "UN Chief says there is no military solution in Syria"
+sent = st.text_area("Input Your Text Here", default_value, height = 275)
 
 if st.button("Run"):
     with st.spinner("Working Hard..."):
-        if src_lang_selection == "English":
-            src_lang = "eng_Latn"
-            tgt_lang = "vie_Latn"
-        else:
-            src_lang = "vie_Latn"
-            tgt_lang = "eng_Latn"
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=src_lang, tgt_lang=tgt_lang)
         trans_text = translator(sent)[0]["translation_text"]
         st.write(trans_text)