Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -142,10 +142,8 @@ class KBVQA():
         return p
        
 
-    def generate_answer(self, question, image):
-        img = copy.deepcopy(image)
-        st.write('image being detcted')
-        st.image(img)
+    def generate_answer(self, question, caption, detected_objects_str,):
+        
         caption = self.get_caption(img)
         image_with_boxes, detected_objects_str = self.detect_objects(img)
         st.write(detected_objects_str)