fix code

app.py

@@ -22,7 +22,7 @@ def create_model():
   # input_layer = Input(shape=(input_ids.shape[1],))
   # embedding_layer = Embedding(max_seq_length+ 1, 200, input_length=max_seq_length)(input_layer)
 
-  inputs = tf.keras.layers.Input(shape=(input_ids.shape[1],), dtype=tf.int32)
+  inputs = tf.keras.layers.Input(shape=(max_seq_length,), dtype=tf.int32)
   embedding_layer = transformer_model(inputs)[0]
 
   flatten_layer = Flatten()(embedding_layer)