Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -249,11 +249,6 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False, force_reload:
     Returns:
         KBVQA: An instance of the KBVQA model ready for inference.
     """
-    
-    free_gpu_resources()
-    kbvqa = KBVQA()
-    kbvqa.detection_model = st.session_state.detection_model
-    # Progress bar for model loading
   
     if force_reload:
         loading_message = 'Force Reloading model.. this should take no more than a few minutes!'
@@ -262,9 +257,14 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False, force_reload:
         except:
             free_gpu_resources()
             pass
-
         free_gpu_resources()
+        
     else: loading_message = 'Looading model.. this should take no more than a few minutes!'
+
+    free_gpu_resources()
+    kbvqa = KBVQA()
+    kbvqa.detection_model = st.session_state.detection_model
+    # Progress bar for model loading
     
     with st.spinner(loading_message):
         if not only_reload_detection_model: