Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -172,7 +172,7 @@ class KBVQA:
         """
         Forces a reload of all models, freeing up GPU resources. This method deletes the current models and calls `free_gpu_resources`.
         """
-        
+        st.write('UUUUUUU')
         free_gpu_resources()
         if self.kbvqa_model is not None:
             del self.kbvqa_model
@@ -274,6 +274,7 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False, force_reload:
     # Progress bar for model loading
   
     if force_reload:
+        st.write('FFFFFFFFF')
         self.delete_model()
         loading_message = 'Force Reloading model.. this should take no more than a few minutes!'
     else: loading_message = 'Looading model.. this should take no more than a few minutes!'