Demosthene-OR commited on
Commit
e9ab91e
1 Parent(s): 2eb8138

Update main_dl.py

Browse files
Files changed (1) hide show
  1. main_dl.py +2 -2
main_dl.py CHANGED
@@ -73,7 +73,7 @@ async def decode_sequence_rnn(input_sentence, src, tgt):
73
  decoded_sentence += " " + sampled_token
74
  if sampled_token == "[end]":
75
  break
76
- return decoded_sentence[8:-6]
77
 
78
  # ===== Enf of Keras ====
79
 
@@ -211,7 +211,7 @@ async def decode_sequence_tranf(input_sentence, src, tgt):
211
  decoded_sentence += " " + sampled_token
212
  if sampled_token == "[end]":
213
  break
214
- return decoded_sentence[8:-6]
215
 
216
  # ==== End Transforformer section ====
217
 
 
73
  decoded_sentence += " " + sampled_token
74
  if sampled_token == "[end]":
75
  break
76
+ return await decoded_sentence[8:-6]
77
 
78
  # ===== Enf of Keras ====
79
 
 
211
  decoded_sentence += " " + sampled_token
212
  if sampled_token == "[end]":
213
  break
214
+ return await decoded_sentence[8:-6]
215
 
216
  # ==== End Transforformer section ====
217