Update load_model.py

load_model.py

@@ -4,7 +4,7 @@ import tensorflow as tf
 from config import config
 
 
-def load_model(en_emb_matrix, de_emb_matrix, model_path, config):
+def load_transformer(en_emb_matrix, de_emb_matrix, model_path, config):
     # Initialize and rebuild your Transformer model
     # (Make sure to replace '...' with actual parameters)
     model = Transformer(
@@ -24,4 +24,9 @@ def load_model(en_emb_matrix, de_emb_matrix, model_path, config):
 def load_sp_model(path_en,path_ur):
     sp_model_en = tf_text.SentencepieceTokenizer(model=tf.io.gfile.GFile(path_en, 'rb').read(),add_bos=True,add_eos=True)
     sp_model_ur = tf_text.SentencepieceTokenizer(model=tf.io.gfile.GFile(path_ur, 'rb').read(),reverse=True,add_bos=True,add_eos=True)
-    return sp_model_en, sp_model_ur
+    return sp_model_en, sp_model_ur
+
+def load_emb(emb_path):
+    with h5py.File(emb_path, 'r') as hf:
+        embedding_matrix = hf['embeddings'][:]
+    return embedding_matrix