Update app.py
Browse files
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.
|
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)
|