Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -110,7 +110,9 @@ class KBVQA():
         return p
        
 
-    def generate_answer(self, question, caption, detected_objects_str):
+    def generate_answer(self, question, image):
+        caption = self.get_caption(image)
+        image_with_boxes, detected_objects_str = self.detect_objects(image)
         prompt = self.format_prompt(question, caption=caption, objects=detected_objects_str)
         num_tokens = len(self.kbvqa_tokenizer.tokenize(prompt))
         if num_tokens > self.max_context_window: