Update my_model/captioner/image_captioning.py

my_model/captioner/image_captioning.py

@@ -66,7 +66,10 @@ class ImageCaptioningModel:
 
     def generate_caption(self, image_path):
 
-        image = Image.open(image_path)
+        if isinstance(image, str) or isinstance(image, io.IOBase):
+        # If it's a file path or file-like object, open it as a PIL Image
+            image = Image.open(image_path)
+            
         image = self.resize_image(image)
         inputs = self.processor(image, self.prompt, return_tensors="pt").to("cuda", self.torch_dtype)
         outputs = self.model.generate(**inputs, min_length=self.min_length, max_new_tokens=self.max_new_tokens)