Kumarkishalaya commited on
Commit
f73c3a8
1 Parent(s): 88a55f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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(lstm_model_path)
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)