Update app.py (#4)

- Update app.py (cec9e44158e28dd2781d257f02d0646a74bca584)


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

app.py

@@ -4,15 +4,8 @@ from tensorflow.keras.models import load_model
 import pickle
 import json
 import keras
+from huggingface_hub import hf_hub_download
 
-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"
@@ -26,7 +19,12 @@ 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 = model_lstm
+repo_id = "Kumarkishalaya/lstm-eng-to-hin"
+lstm_filename = "seq2seq_model.h5" 
+lstm_model_path = hf_hub_download(repo_id=repo_id, filename=lstm_filename)
+
+model_lstm = load_model(lstm_model_path)
+
 with open('eng_tokenizer.pkl', 'rb') as file:
     eng_tokenizer = pickle.load(file)
 with open('hin_tokenizer.pkl', 'rb') as file: