Spaces:
Paused
Paused
Update app.py
Browse files
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(
|
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
|