Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -136,7 +136,7 @@ class KBVQA:
         """
         st.write('detect func', self.detection_confidence)
         image = self.detector.process_image(img)
-        detected_objects_string, detected_objects_list = self.detector.detect_objects(image, threshold=self.detection_confidence)
+        detected_objects_string, detected_objects_list = self.detector.detect_objects(image, threshold=st.session_state['confidence_level'])
         image_with_boxes = self.detector.draw_boxes(img, detected_objects_list)
         return image_with_boxes, detected_objects_string