Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -48,7 +48,7 @@ class KBVQA():
 
 
     def load_caption_model(self):
-        self.captioner = ImageCaptioningModel(model_type='i_blip')
+        self.captioner = ImageCaptioningModel()
         self.captioner.load_model()
 
     def get_caption(self, img):
@@ -145,11 +145,11 @@ def prepare_kbvqa_model(detection_model):
     # Progress bar for model loading
     with st.spinner('Loading models...'):
         progress_bar = st.progress(0)
-        kbvqa.load_fine_tuned_model()
+        kbvqa.load_detector(detection_model)
         progress_bar.progress(33)
         kbvqa.load_caption_model()
         progress_bar.progress(66)
-        kbvqa.load_detector(detection_model)  # Replace with your model
+        kbvqa.load_fine_tuned_model()
         progress_bar.progress(100)
     
     if kbvqa.all_models_loaded: