Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -23,6 +23,7 @@ class KBVQA():
         self.captioner = None
         self.detector = None
         self.detection_model = None
+        self.detecttion_confidence = None
         self.kbvqa_model = None
         self.access_token = os.getenv("HUGGINGFACE_TOKEN")
       #  self.kbvqa_model_loaded = self.all_models_loaded() 
@@ -63,9 +64,9 @@ class KBVQA():
         self.detector = ObjectDetector()
         self.detector.load_model(model)
 
-    def detect_objects(self, img, threshold=0.2):
+    def detect_objects(self, img):
         image = self.detector.process_image(img)
-        detected_objects_string, detected_objects_list = self.detector.detect_objects(image, threshold=threshold)
+        detected_objects_string, detected_objects_list = self.detector.detect_objects(image, threshold=self.detection_confidence)
         image_with_boxes = self.detector.draw_boxes(img, detected_objects_list)
         return image_with_boxes, detected_objects_string