I am facing trouble and confusion as to how can i save and deploy this model for a web app here is my model architecture code:
It has first 2 layers from tensorflow hub (bert model)
I face trouble everytime I load it from .h5 version saying that Hub.KerasLayers are custom and I need to define them.
Is there any workaround for this?
# Load BERT preprocessor and encoder
bert_preprocess = hub.KerasLayer("https://tfhub.dev/tensorflow/bert_en_uncased_preprocess/3")
bert_encoder = hub.KerasLayer("https://tfhub.dev/tensorflow/bert_en_uncased_L-12_H-768_A-12/4")
# Define input layer
text_input = tf.keras.layers.Input(shape=(), dtype=tf.string, name='text')
# Preprocess text using BERT
preprocessed_text = bert_preprocess(text_input)
# Encode preprocessed text using BERT
outputs = bert_encoder(preprocessed_text)
reshaped_output = tf.expand_dims(outputs['pooled_output'], axis=1)
# LSTM layer
lstm_layer = tf.keras.layers.LSTM(64, name="lstm")(reshaped_output)
# Neural network layers
l = tf.keras.layers.Dropout(0.1, name="dropout")(outputs['pooled_output'])
l=tf.keras.layers.Dense(32,activation='relu',name='hiddenn')(l)
l=tf.keras.layers.Dense(8,activation='relu',name='hiddennn')(l)
l = tf.keras.layers.Dense(1, activation='sigmoid', name="output")(l)
# Construct final model
model = tf.keras.Model(inputs=[text_input], outputs=[l])
Should I change the custom bert layers?
error i am getting during deployment:
Uncaught (in promise) Error: Unknown layer: KerasLayer. This may be due to one of the following reasons:
1. The layer is defined in Python, in which case it needs to be ported to TensorFlow.js or your JavaScript code.
2. The custom layer is defined in JavaScript, but is not registered properly with tf.serialization.registerClass().
at Vse (generic_utils.js:243:13)
at rle (serialization.js:31:10)
at s (container.js:1264:11)
at value (container.js:1292:7)
at Vse (generic_utils.js:278:11)
at rle (serialization.js:31:10)
at models.js:308:7
at h (tf.min.js:17:2100)
at Generator.<anonymous> (tf.min.js:17:3441)
at Generator.next (tf.min.js:17:2463)