Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -152,7 +152,7 @@ def load_scripted_model(path):
|
|
152 |
@st.cache_resource
|
153 |
def preload_models():
|
154 |
return {
|
155 |
-
"Bi-Directional LSTM": load_scripted_model("
|
156 |
"Gated Recurrent Unit (GRU)": load_model_from_disk("best_gru_model.pth", model_type="GRU")
|
157 |
}
|
158 |
|
|
|
152 |
@st.cache_resource
|
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 |
|