Update app.py (#3)

- Update app.py (4af16022d9721e838c4d9d0ee922fb58eec5e09f)


Co-authored-by: Kumar Kishalaya <[email protected]>

app.py

@@ -2,6 +2,17 @@ import gradio as gr
 from transformers import AutoTokenizer, TFAutoModelForSeq2SeqLM
 from tensorflow.keras.models import load_model
 import pickle
+import json
+import keras
+
+custom_objects = {
+    'LSTM': keras.layers.LSTM,
+}
+# Load model
+model_lstm = load_model('seq2seq_model.h5', custom_objects=custom_objects, compile=False)
+
+# Recompile the model
+model_lstm.compile(optimizer='adam', loss='sparse_categorical_crossentropy')
 
 # Define the model repository and tokenizer checkpoint
 model_checkpoint = "himanishprak23/neural_machine_translation"
@@ -15,7 +26,7 @@ tokenizer_nmt = AutoTokenizer.from_pretrained(tokenizer_checkpoint)
 model_nmt = TFAutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 
 # Loading models, tokenizer & variables for trained LSTM translation model.
-model_lstm = load_model('seq2seq_model.h5')
+model_lstm = model_lstm
 with open('eng_tokenizer.pkl', 'rb') as file:
     eng_tokenizer = pickle.load(file)
 with open('hin_tokenizer.pkl', 'rb') as file: