Update README.md

README.md

@@ -65,8 +65,8 @@ MAX_SEQUENCE_LENGTH = 200
 model = tf.keras.models.load_model('RETO-SENTIMENT.h5', custom_objects={'AttentionLayer': AttentionLayer})
 
 class AttentionLayer(tf.keras.layers.Layer):
-    def __init__(self):
-        super(AttentionLayer, self).__init__()
+    def __init__(self, **kwargs):  
+        super(AttentionLayer, self).__init__(**kwargs)
 
     def call(self, inputs):
         attention_weights = tf.nn.softmax(tf.reduce_sum(inputs, axis=-1), axis=-1)