Update app.py

app.py

@@ -4,7 +4,7 @@ from keras.layers import Input, Dense, Embedding, MultiHeadAttention
 from keras.layers import Dropout, LayerNormalization
 from keras.models import Model
 from keras.utils import pad_sequences
-from tensorflow.keras import layers
+from keras import layers
 import numpy as np
 import logging
 logging.getLogger('tensorflow').setLevel(logging.ERROR)