Update modeling_GOT.py

modeling_GOT.py

@@ -716,7 +716,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
         return processed_images
 
 
-    def chat_crop(self, tokenizer, image_file, render=False, save_render_file=None, print_prompt=False, gradio_input=False):
+    def chat_crop(self, tokenizer, image_file, ocr_type, render=False, save_render_file=None, print_prompt=False, gradio_input=False):
         # Model
         self.disable_torch_init()
         multi_page=False
@@ -751,7 +751,10 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             # print("len ll: ", ll)
 
         else:
-            qs = 'OCR with format upon the patch reference: '
+            if ocr_type == 'format':
+                qs = 'OCR with format upon the patch reference: '
+            else:
+                qs = 'OCR upon the patch reference: '
             if gradio_input:
                 img = image_file.copy()
             else: