florentgbelidji
/

blip_captioning

image-captioning

endpoints-template

Inference Endpoints

Model card Files Files and versions Community

florentgbelidji HF staff commited on Aug 17, 2022

Commit

f0b2295

·

1 Parent(s): a77885d

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -14,7 +14,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 class PreTrainedPipeline():
     def __init__(self, path=""):
         # load the optimized model
-        self.model_path = 'model_large_caption.pth'
         self.model = blip_decoder(
             pretrained=self.model_path,
             image_size=384,

 class PreTrainedPipeline():
     def __init__(self, path=""):
         # load the optimized model
+        self.model_path = os.path.join(path,'model_large_caption.pth')
         self.model = blip_decoder(
             pretrained=self.model_path,
             image_size=384,