Update app.py
Browse files
app.py
CHANGED
@@ -26,15 +26,15 @@ tokenizer_nmt = AutoTokenizer.from_pretrained(tokenizer_checkpoint)
|
|
26 |
model_nmt = TFAutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
|
27 |
|
28 |
# Loading models, tokenizer & variables for trained LSTM translation model.
|
29 |
-
repo_id = "Kumarkishalaya/lstm-eng-to-hin"
|
30 |
-
lstm_filename = "seq2seq_model.keras"
|
31 |
|
32 |
|
33 |
# Re-download the file
|
34 |
-
lstm_model_path = hf_hub_download(repo_id=repo_id, filename=lstm_filename, force_download=True)
|
35 |
|
36 |
|
37 |
-
model_lstm = load_model(
|
38 |
|
39 |
with open('eng_tokenizer.pkl', 'rb') as file:
|
40 |
eng_tokenizer = pickle.load(file)
|
|
|
26 |
model_nmt = TFAutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
|
27 |
|
28 |
# Loading models, tokenizer & variables for trained LSTM translation model.
|
29 |
+
#repo_id = "Kumarkishalaya/lstm-eng-to-hin"
|
30 |
+
#lstm_filename = "seq2seq_model.keras"
|
31 |
|
32 |
|
33 |
# Re-download the file
|
34 |
+
#lstm_model_path = hf_hub_download(repo_id=repo_id, filename=lstm_filename, force_download=True)
|
35 |
|
36 |
|
37 |
+
model_lstm = load_model('seq2seq_model.h5')
|
38 |
|
39 |
with open('eng_tokenizer.pkl', 'rb') as file:
|
40 |
eng_tokenizer = pickle.load(file)
|