Puyush commited on
Commit
7fe476e
·
verified ·
1 Parent(s): 8b0a550

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.h5", custom_objects={"CTCLayer": CTCLayer})
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