Spaces:

RishabA
/

spanish-to-english

Sleeping

RishabA commited on Apr 8

Commit

923823a

verified ·

1 Parent(s): f8dfaca

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -623,5 +623,5 @@ def translate_sentence(
         if pred_token == vocab_trg.stoi["<eos>"]:
             break
         trg_indexes.append(pred_token)
-    trg_tokens = vocab_trg.lookup_tokens(trg_indexes)
     return " ".join(trg_tokens)

         if pred_token == vocab_trg.stoi["<eos>"]:
             break
         trg_indexes.append(pred_token)
+    trg_tokens = vocab_trg.lookup_tokens(trg_indexes[1:])
     return " ".join(trg_tokens)