Update my_model/captioner/image_captioning.py

my_model/captioner/image_captioning.py

@@ -6,7 +6,8 @@ from transformers import InstructBlipProcessor, InstructBlipForConditionalGenera
 import bitsandbytes
 import accelerate
 from my_model.captioner import captioning_config as config
-
+from my_model.utilities import free_gpu_resources
+    
 
 class ImageCaptioningModel:
     def __init__(self):
@@ -76,6 +77,8 @@ def get_caption(img):
     captioner.load_model()
     caption = captioner.generate_caption(img)
 
+    free_gpu_resources()
+
     return caption