Update app.py

app.py

@@ -9,7 +9,7 @@ model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True,
 model = model.eval().cuda()
 
 @spaces.GPU
-def run_GOT(image_array, got_mode, ocr_box="", ocr_color=""):
+def run_GOT(image_array, got_mode, fine_grained_mode="", ocr_color="", ocr_box=""):
     image = image_array
     if got_mode == "plain texts OCR":
         res = model.chat(tokenizer, image, ocr_type='ocr')
@@ -135,7 +135,7 @@ with gr.Blocks() as demo:
     
     submit_button.click(
         run_GOT,
-        inputs=[image_input, task_dropdown, box_input, color_dropdown],
+        inputs=[image_input, task_dropdown, fine_grained_dropdown, color_dropdown, box_input],
         outputs=[ocr_result, html_result]
     )