Mattral commited on
Commit
7a745eb
·
verified ·
1 Parent(s): 0892dcd

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("KhunPop/Gardening")
23
- model = AutoModelForCausalLM.from_pretrained("unsloth/gemma-2-2b)
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("KhunPop/Gardening")
23
+ model = AutoModelForCausalLM.from_pretrained("TheSheBots/UrbanGardening")
24
  # Store the model and tokenizer in session state
25
  st.session_state.tokenizer = tokenizer
26
  st.session_state.model = model