yashwan2003
/

engtokantranslation

machine-translation

Model card Files Files and versions

yashwan2003 commited on Feb 8

Commit

1025f82

·

verified ·

1 Parent(s): 608ece1

Update main.py

Files changed (1) hide show

main.py +1 -1

main.py CHANGED Viewed

@@ -689,7 +689,7 @@ def greedy_decode(
     model = model.to(device)
     # Encode source sentence
-    src_ids = source_tokenizer.encode(src_sentence)
     src_ids = src_ids[:max_length]
     src_padded = src_ids + [vocab_info['src_pad_idx']] * (max_length - len(src_ids))
     src = torch.tensor([src_padded], dtype=torch.long).to(device)

     model = model.to(device)
     # Encode source sentence
+    src_ids = source_tokenizer.encode(src_sentence.lower()).ids
     src_ids = src_ids[:max_length]
     src_padded = src_ids + [vocab_info['src_pad_idx']] * (max_length - len(src_ids))
     src = torch.tensor([src_padded], dtype=torch.long).to(device)