slight mods

app.py

@@ -12,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 def load_model(model_name):
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     return model
-    
+
 model = load_model("facebook/nllb-200-distilled-600M")
 
 default_value = "UN Chief says there is no military solution in Syria"
@@ -21,15 +21,14 @@ 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"
-
 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)