Mattral commited on
Commit
4e08a94
·
verified ·
1 Parent(s): 7a745eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def load_model():
19
  if "tokenizer" in st.session_state and "model" in st.session_state:
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
 
19
  if "tokenizer" in st.session_state and "model" in st.session_state:
20
  return st.session_state.tokenizer, st.session_state.model
21
  else:
22
+ tokenizer = AutoTokenizer.from_pretrained("TheSheBots/UrbanGardening")
23
  model = AutoModelForCausalLM.from_pretrained("TheSheBots/UrbanGardening")
24
  # Store the model and tokenizer in session state
25
  st.session_state.tokenizer = tokenizer