Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -168,7 +168,7 @@ class KBVQA:
         
         return self.kbvqa_model is not None and self.captioner is not None and self.detector is not None
 
-    def force_reload_model(self):
+    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`.
         """
@@ -186,7 +186,6 @@ class KBVQA:
             
         free_gpu_resources()
         
-        return prepare_kbvqa_model(only_reload_detection_model=False, force_reload=True)
 
 
     def format_prompt(self, current_query: str, history: Optional[str] = None, sys_prompt: Optional[str] = None, caption: str = None, objects: Optional[str] = None) -> str:
@@ -275,6 +274,7 @@ def prepare_kbvqa_model(only_reload_detection_model: bool = False, force_reload:
     # Progress bar for model loading
     with kbvqa.col1:
         if force_reload:
+            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!'