danielle2003 commited on
Commit
121d7ab
·
verified ·
1 Parent(s): d762f15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -153,7 +153,7 @@ def load_scripted_model(path):
153
  def preload_models():
154
  return {
155
  "Bi-Directional LSTM": load_scripted_model("bilstm_scriptes.pt"),
156
- "Gated Recurrent Unit (GRU)": load_model_from_disk("best_gru_model.pth", model_type="GRU")
157
  }
158
 
159
  MODELS = preload_models()
@@ -220,7 +220,7 @@ with st.sidebar:
220
  ticker = st.text_input("Stock Ticker", st.session_state.get('last_ticker', "AMZN"), disabled=True).upper()
221
  model_type = st.selectbox(
222
  "Prediction Model",
223
- ("Bi-Directional LSTM", "Gated Recurrent Unit (GRU)"),
224
  key="model_choice",
225
  help="Select the neural network architecture for prediction."
226
  )
 
153
  def preload_models():
154
  return {
155
  "Bi-Directional LSTM": load_scripted_model("bilstm_scriptes.pt"),
156
+ "Gated Recurrent Unit": load_model_from_disk("best_gru_model.pth", model_type="GRU")
157
  }
158
 
159
  MODELS = preload_models()
 
220
  ticker = st.text_input("Stock Ticker", st.session_state.get('last_ticker', "AMZN"), disabled=True).upper()
221
  model_type = st.selectbox(
222
  "Prediction Model",
223
+ ("Bi-Directional LSTM", "Gated Recurrent Unit"),
224
  key="model_choice",
225
  help="Select the neural network architecture for prediction."
226
  )