Update app.py

app.py

@@ -15,7 +15,7 @@ model_keras_path = "./model.keras"
 model_h5_path = "./model.h5"
 
 # model = tf.keras.models.load_model(model_keras_path, custom_objects={'KerasLayer': hub.KerasLayer})
-model = tf.keras.models.load_model(model_h5_path)
+model = tf.keras.models.load_model(model_h5_path, {"KerasLayer": hub.KerasLayer})
 # model.compile(optimizer=Adam(learning_rate=0.0001),
 #               loss='binary_crossentropy',
 #               metrics=['accuracy'])