Update main.py

main.py

@@ -56,7 +56,7 @@ body {{
 )
 
 # -----------------------------
-# LOAD MODEL WITH LOADER
+# LOAD MODEL (NO LOADER HERE)
 # -----------------------------
 BASE_MODEL = "facebook/nllb-200-distilled-600M"
 LORA_REPO = "junaid17/nllb-kurdish-lora"
@@ -64,13 +64,9 @@ LORA_REPO = "junaid17/nllb-kurdish-lora"
 
 @st.cache_resource
 def load_model():
-    with st.spinner(""):
-        st.markdown("<div class='dots-loader'><div></div><div></div><div></div></div>", unsafe_allow_html=True)
-
-        tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
-        base = AutoModelForSeq2SeqLM.from_pretrained(BASE_MODEL)
-        model = PeftModel.from_pretrained(base, LORA_REPO)
-
+    tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
+    base = AutoModelForSeq2SeqLM.from_pretrained(BASE_MODEL)
+    model = PeftModel.from_pretrained(base, LORA_REPO)
     return model, tokenizer
 
 
@@ -111,21 +107,16 @@ with colTheme:
 with col2:
     tgt_lang = st.selectbox("To:", ["Kurdish", "English"])
 
-
-# Handle swap
+# Swap
 if swap:
     src_lang, tgt_lang = tgt_lang, src_lang
-    st.session_state["LANG1"] = src_lang
-    st.session_state["LANG2"] = tgt_lang
     st.rerun()
 
-# Handle dark/light switch
+# Toggle theme
 if toggle:
     st.session_state["theme"] = "dark" if THEME == "light" else "light"
     st.rerun()
 
-
-# Language mapping
 lang_codes = {
     "English": "eng_Latn",
     "Kurdish": "ckb_Arab"
@@ -136,23 +127,24 @@ tgt_code = lang_codes[tgt_lang]
 
 text = st.text_area("Enter text:", height=180)
 
+
 # -----------------------------
-# TRANSLATE BUTTON
+# TRANSLATE BUTTON (with loader)
 # -----------------------------
 if st.button("Translate"):
     if not text.strip():
         st.warning("Please enter some text.")
     else:
-        # Show 3-dot loader ONLY during translation
-        loader = st.markdown("<div class='dots-loader'><div></div><div></div><div></div></div>", unsafe_allow_html=True)
+        loader = st.markdown(
+            "<div class='dots-loader'><div></div><div></div><div></div></div>",
+            unsafe_allow_html=True
+        )
 
         output = translate(text, src_code, tgt_code)
 
-        # Remove loader
         loader.empty()
 
         st.subheader("Output")
         st.markdown(f"<div class='output-box'>{output}</div>", unsafe_allow_html=True)
 
-        # Copy button
         st.button("Copy Output", on_click=lambda: st.session_state.update({"copied": output}))