Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -167,24 +167,6 @@ class KBVQA:
         """
         
         return self.kbvqa_model is not None and self.captioner is not None and self.detector is not None
-
-    def delete_model(self):
-        """
-        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
-            free_gpu_resources()
-        if self.captioner is not None:
-            del self.captioner
-            free_gpu_resources()
-        if self.detector is not None:
-            del self.detector
-            free_gpu_resources()
-            
-        free_gpu_resources()
         
 
 
@@ -274,8 +256,6 @@ 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!'