Create imageCaptioning.py (#3)

- Create imageCaptioning.py (2b996da8d9dc6b4c3f1b546d7c965e53a6936607)


Co-authored-by: Khan <[email protected]>

src/models/imageCaptioning.py

@@ -0,0 +1,15 @@
+from PIL import Image
+from transformers import BlipProcessor , BlipForConditionalGeneration
+import torch
+
+
+processor = BlipProcessor.from_pretrained("src/models/Caption")
+model = BlipForConditionalGeneration.from_pretrained("src/models/Caption")
+
+
+def generateCaption(image_path):
+    image = Image.open(image_path).convert("RGB")
+    inputs = processor(images = image , return_tensors="pt")
+    output = model.generate(**inputs)
+    caption = processor.decode(output[0], skip_special_tokens = True)
+    return caption