Update app.py

app.py

@@ -15,7 +15,11 @@ HF_TOKEN = st.secrets["HF_TOKEN"]  # Put your HF token in Streamlit secrets
 st.title("🧠 Niche Trainer with Push to HF")
 
 # ---------- Load model once ----------
-@st.cache_resource(show_spinner=True)
+# DO NOT load on startup
+# tokenizer, model = load_model()  <- remove from top
+
+# Instead, load when first used:
+@st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
     model = AutoModelForCausalLM.from_pretrained(