Spaces:
Sleeping
Sleeping
Hardik commited on
Commit ·
525fd77
1
Parent(s): eafed38
LSTM: try keras.saving.load_model first
Browse files
ml_service/app/services/model_loader.py
CHANGED
|
@@ -108,6 +108,7 @@ class ModelManager:
|
|
| 108 |
last_lstm_error = ""
|
| 109 |
|
| 110 |
for loader_fn, name in [
|
|
|
|
| 111 |
(lambda p: keras.models.load_model(p), "keras"),
|
| 112 |
(lambda p: tf.keras.models.load_model(p, compile=False), "tf.keras"),
|
| 113 |
]:
|
|
|
|
| 108 |
last_lstm_error = ""
|
| 109 |
|
| 110 |
for loader_fn, name in [
|
| 111 |
+
(lambda p: keras.saving.load_model(p), "keras_saving"),
|
| 112 |
(lambda p: keras.models.load_model(p), "keras"),
|
| 113 |
(lambda p: tf.keras.models.load_model(p, compile=False), "tf.keras"),
|
| 114 |
]:
|