Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -141,6 +141,8 @@ class KBVQA():
        
 
     def generate_answer(self, question, image):
+        st.write('image being dected')
+        st.image(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)