Update app.py

app.py

@@ -8,6 +8,7 @@ import torch.nn as nn
 from transformers import Blip2Processor, Blip2ForConditionalGeneration, InstructBlipProcessor, InstructBlipForConditionalGeneration
 from my_model.object_detection import detect_and_draw_objects
 from my_model.captioner.image_captioning import get_caption
+from my_model.utilities import free_gpu_resources
 
 
     
@@ -38,11 +39,13 @@ image = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
 question = st.text_input("Enter your question about the image:")
 
 if st.button('Generate Caption'):
+    free_gpu_resources()
     if image is not None:
         # Display the image
         st.image(image, use_column_width=True)
         caption = get_caption(image)
         st.write(caption)
+        free_gpu_resources()
 
     else:
         st.write("Please upload an image and enter a question.")