Update app.py

app.py

@@ -20,7 +20,7 @@ def load_model():
             return st.session_state.tokenizer, st.session_state.model
         else:
             tokenizer = AutoTokenizer.from_pretrained("TheSheBots/UrbanGardening", use_auth_token=HF_TOKEN)
-            model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it", use_auth_token=HF_TOKEN)
+            model = AutoModelForCausalLM.from_pretrained("unsloth/gemma-2-2b-bnb-4bit", use_auth_token=HF_TOKEN)
             # Store the model and tokenizer in session state
             st.session_state.tokenizer = tokenizer
             st.session_state.model = model