SOS error again

app/image_captioning_service.py

@@ -301,7 +301,7 @@ class ImageCaptioningModel(torch.nn.Module):
             img_features = img_features.unsqueeze(1)
             
             # Start with < SOS > token
-            current_ids = torch.tensor([[vocab.word2idx['< SOS >']]], dtype=torch.long).to(image.device)
+            current_ids = torch.tensor([[vocab.word2idx['<SOS>']]], dtype=torch.long).to(image.device)
             
             # Generate words one by one
             result_caption = []