Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import torch
+import copy
 import os
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from typing import Optional
@@ -141,10 +142,12 @@ 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)
+        img = copy.deepcopy(image)
+        st.write('image being detcted')
+        st.image(img)
+        caption = self.get_caption(img)
+        image_with_boxes, detected_objects_str = self.detect_objects(img)
+        st.write(detected_objects_str)
         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: