Missing self

pipeline.py

@@ -17,7 +17,7 @@ class PreTrainedPipeline():
         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.eval()
-        self.model = model.to(device)
+        self.model = self.model.to(device)
         
         image_size = 384
         self.transform = transforms.Compose([
@@ -43,8 +43,8 @@ class PreTrainedPipeline():
 
         # decode base64 image to PIL
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
-        image = transform(image).unsqueeze(0).to(device)   
+        image = self.transform(image).unsqueeze(0).to(device)   
         with torch.no_grad():
             caption = self.model.generate(image, sample=True, top_p=0.9, max_length=20, min_length=5)
         # postprocess the prediction
-        return caption
+        return caption