Update pipeline.py (#1)

- Update pipeline.py (8085a1b108c107456579c23e3ea0fc5ecd33103c)


Co-authored-by: Philipp Schmid <[email protected]>

pipeline.py

@@ -15,7 +15,7 @@ class PreTrainedPipeline():
     def __init__(self, path=""):
         # load the optimized model
         self.model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth'
-        self.model = blip_decoder(pretrained=self.model_url, image_size=384, vit='large')
+        self.model = blip_decoder(pretrained=self.model_url, image_size=384, vit='large',med_config=os.path.join(path, 'configs/med_config.json'))
         self.model.eval()
         self.model = self.model.to(device)