Removing the load_img function before processor
Browse files- utils/caption_utils.py +1 -2
utils/caption_utils.py
CHANGED
@@ -11,8 +11,7 @@ class ImageCaptioning:
|
|
11 |
self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
12 |
self.model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
|
13 |
|
14 |
-
def get_caption(self,
|
15 |
-
image = load_image(image_path)
|
16 |
|
17 |
# Preprocessing the Image
|
18 |
img = self.processor(image, return_tensors="pt").to(device)
|
|
|
11 |
self.processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
|
12 |
self.model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base").to(device)
|
13 |
|
14 |
+
def get_caption(self, image):
|
|
|
15 |
|
16 |
# Preprocessing the Image
|
17 |
img = self.processor(image, return_tensors="pt").to(device)
|