Update src/utils/captioning_inference.py

src/utils/captioning_inference.py

@@ -4,12 +4,12 @@ import sentencepiece as spm
 import torch
 import yaml
 
-from dataset.build_vocab import build_vocab, tokenizer
-from models.resnet18 import EncoderResnet18
-from models.swin import EncoderSwinTiny
-from models.transformer import DecoderTransformer
-from models.vit import EncoderViTB16
-from transforms.image_transform import get_caption_transform
+from src.dataset.build_vocab import build_vocab, tokenizer
+from src.models.resnet18 import EncoderResnet18
+from src.models.swin import EncoderSwinTiny
+from src.models.transformer import DecoderTransformer
+from src.models.vit import EncoderViTB16
+from src.transforms.image_transform import get_caption_transform
 
 
 def resolve_path(workspace_root, path_value):