Update my_model/state_manager.py

my_model/state_manager.py

@@ -162,6 +162,7 @@ class StateManager:
             st.session_state['model_loaded'] = True
             st.session_state['button_label'] = "Reload Model"
             free_gpu_resources()
+            free_gpu_resources()
             
         except Exception as e:
             st.error(f"Error loading model: {e}")
@@ -202,7 +203,9 @@ class StateManager:
             try:
                 del st.session_state['kbvqa']
                 free_gpu_resources()
+                free_gpu_resources()
             except:
+                free_gpu_resources()
                 free_gpu_resources()
                 pass
             
@@ -314,6 +317,9 @@ class StateManager:
         Returns:
             tuple: A tuple containing the generated caption, detected objects string, and image with bounding boxes.
         """
+
+        free_gpu_resources()
+        free_gpu_resources()
         img = copy.deepcopy(image)
         caption = st.session_state['kbvqa'].get_caption(img)
         image_with_boxes, detected_objects_str = st.session_state['kbvqa'].detect_objects(img)