Kumarkishalaya commited on
Commit
1b432a6
·
verified ·
1 Parent(s): 835a881

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ model_nmt = TFAutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
20
 
21
  # Loading models, tokenizer & variables for trained LSTM translation model.
22
  repo_id = "Kumarkishalaya/lstm-eng-to-hin"
23
- lstm_filename = "seq2seq_model.h5"
24
  lstm_model_path = hf_hub_download(repo_id=repo_id, filename=lstm_filename)
25
 
26
  model_lstm = load_model(lstm_model_path)
 
20
 
21
  # Loading models, tokenizer & variables for trained LSTM translation model.
22
  repo_id = "Kumarkishalaya/lstm-eng-to-hin"
23
+ lstm_filename = "seq2seq_model.keras"
24
  lstm_model_path = hf_hub_download(repo_id=repo_id, filename=lstm_filename)
25
 
26
  model_lstm = load_model(lstm_model_path)