Mattral commited on
Commit
7a3a8fa
·
verified ·
1 Parent(s): 6dc632c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def load_model():
20
  return st.session_state.tokenizer, st.session_state.model
21
  else:
22
  tokenizer = AutoTokenizer.from_pretrained("TheSheBots/UrbanGardening", use_auth_token=HF_TOKEN)
23
- model = AutoModelForCausalLM.from_pretrained("CopyleftCultivars/Gemma2B-NaturalFarmerV1", use_auth_token=HF_TOKEN)
24
  # Store the model and tokenizer in session state
25
  st.session_state.tokenizer = tokenizer
26
  st.session_state.model = model
 
20
  return st.session_state.tokenizer, st.session_state.model
21
  else:
22
  tokenizer = AutoTokenizer.from_pretrained("TheSheBots/UrbanGardening", use_auth_token=HF_TOKEN)
23
+ model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it", use_auth_token=HF_TOKEN)
24
  # Store the model and tokenizer in session state
25
  st.session_state.tokenizer = tokenizer
26
  st.session_state.model = model