316usman commited on
Commit
f236b24
·
1 Parent(s): a8fabcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ if "tokenizer" not in st.session_state:
60
  # Display the "Load Model" button
61
  if not st.session_state["model_loaded"]:
62
  if st.button("Load Model"):
63
- model1, tokenizer1 = load_model(repo_id)
64
  st.session_state["model"] = model1
65
  st.session_state["tokenizer"] = tokenizer1
66
  st.session_state["model_loaded"] = True
 
60
  # Display the "Load Model" button
61
  if not st.session_state["model_loaded"]:
62
  if st.button("Load Model"):
63
+ model1, tokenizer1 = load_model()
64
  st.session_state["model"] = model1
65
  st.session_state["tokenizer"] = tokenizer1
66
  st.session_state["model_loaded"] = True