Update my_model/captioner/image_captioning.py

my_model/captioner/image_captioning.py

@@ -69,6 +69,14 @@ class ImageCaptioningModel:
     def generate_captions_for_multiple_images(self, image_paths):
 
         return [self.generate_caption(image_path) for image_path in image_paths]
+        
+
+def get_caption(img):
+    captioner = ImageCaptioningModel()
+    captioner.load_model()
+    caption = captioner.generate_caption(img)
+
+    return caption
 
 
 if __name__ == "__main__":