Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -257,6 +257,10 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False, force_reload:
   
     if force_reload:
         loading_message = 'Force Reloading model.. this should take no more than a few minutes!'
+        if kbvqa is not None:
+            del kbvqa
+            free_gpu_resources()
+        free_gpu_resources()
     else: loading_message = 'Looading model.. this should take no more than a few minutes!'
     
     with st.spinner(loading_message):
@@ -272,6 +276,7 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False, force_reload:
             free_gpu_resources()
             progress_bar.progress(100)
         else:
+            free_gpu_resources()
             progress_bar = st.progress(0)
             kbvqa.load_detector(kbvqa.detection_model)
             progress_bar.progress(100)