Reaumur commited on
Commit
908aa62
·
verified ·
1 Parent(s): 55cee5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from keras.models import load_model
8
  # Caching the model loading function to optimize performance
9
  @st.cache_resource
10
  def load_model():
11
- model_path = ('captcha.keras', custom_objects={'CustomLayer': CustomLayer}) # Update with the actual model path
12
  return tf.keras.models.load_model(model_path)
13
 
14
  # Load the model
 
8
  # Caching the model loading function to optimize performance
9
  @st.cache_resource
10
  def load_model():
11
+ model_path = "captcha.keras" # Update with the actual model path
12
  return tf.keras.models.load_model(model_path)
13
 
14
  # Load the model