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