Spaces:
Build error
Build error
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 |
|