Update app.py

app.py

@@ -19,8 +19,8 @@ def load_model():
         if "tokenizer" in st.session_state and "model" in st.session_state:
             return st.session_state.tokenizer, st.session_state.model
         else:
-            tokenizer = AutoTokenizer.from_pretrained("TheSheBots/UrbanGardening")
-            model = AutoModelForCausalLM.from_pretrained("TheSheBots/UrbanGardening")
+            tokenizer = AutoTokenizer.from_pretrained("TheSheBots/UrbanGardening", use_auth_token=api_key)
+            model = AutoModelForCausalLM.from_pretrained("TheSheBots/UrbanGardening", use_auth_token=api_key)
             # Store the model and tokenizer in session state
             st.session_state.tokenizer = tokenizer
             st.session_state.model = model