Import fix

blip_decoder.py

@@ -8,8 +8,8 @@
 import warnings
 warnings.filterwarnings("ignore")
 
-from models.vit import VisionTransformer, interpolate_pos_embed
-from models.med import BertConfig, BertModel, BertLMHeadModel
+from vit import VisionTransformer, interpolate_pos_embed
+from med import BertConfig, BertModel, BertLMHeadModel
 from transformers import BertTokenizer
 
 import torch

pipeline.py

@@ -5,7 +5,7 @@ import torch
 import base64
 import os
 from io import BytesIO
-from models.blip_decoder import blip_decoder
+from blip_decoder import blip_decoder
 from torchvision import transforms
 from torchvision.transforms.functional import InterpolationMode