Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -84,7 +84,7 @@ class CTCLayer(layers.Layer):
|
|
84 |
# At test time, just return the computed predictions
|
85 |
return y_pred
|
86 |
|
87 |
-
loaded_model = keras.models.load_model("ocr_model_pred.
|
88 |
loaded_model.load_weights("ocr_model_pred_weights.h5")
|
89 |
max_len = 5
|
90 |
|
|
|
84 |
# At test time, just return the computed predictions
|
85 |
return y_pred
|
86 |
|
87 |
+
loaded_model = keras.models.load_model("ocr_model_pred.keras", custom_objects={"CTCLayer": CTCLayer})
|
88 |
loaded_model.load_weights("ocr_model_pred_weights.h5")
|
89 |
max_len = 5
|
90 |
|