Update modeling_GOT.py

modeling_GOT.py

@@ -484,7 +484,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
         setattr(torch.nn.Linear, "reset_parameters", lambda self: None)
         setattr(torch.nn.LayerNorm, "reset_parameters", lambda self: None)
 
-    def chat(self, tokenizer, image_file, ocr_type, ocr_box='', ocr_color='', render=False, save_render_file=None, print_prompt=False, gradio_input=False, stream_flag = False):
+    def chat(self, tokenizer, image_file, ocr_type, vqa_question='' ,ocr_box='', ocr_color='', render=False, save_render_file=None, print_prompt=False, gradio_input=False, stream_flag = False):
 
         self.disable_torch_init()
 
@@ -505,7 +505,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
         if ocr_type == 'format':
             qs = 'OCR with format: '
         elif ocr_type == 'VQA':
-            qs = 'Answer the following Question :'
+            qs = 'Answer the following Question :' + str(vqa_question)
         else:
             qs = 'OCR: '