Update my_model/KBVQA.py

my_model/KBVQA.py

@@ -257,6 +257,7 @@ class KBVQA:
         prompt = self.format_prompt(question, caption=caption, objects=detected_objects_str)
         num_tokens = len(self.kbvqa_tokenizer.tokenize(prompt))
         self.current_prompt_length = num_tokens
+        trim = False
         if self.current_prompt_length > self.max_context_window:
             trim = True
             st.warning(f"Prompt length is {self.current_prompt_length} which is larger than the maximum context window of LLaMA-2, objects detected with low confidence will be removed one at a time until the prompt length is within the maximum context window ...")