Update app.py

app.py

@@ -10,12 +10,9 @@ model = model.eval().cuda()
 
 @spaces.GPU
 def run_GOT(image_array, got_mode, ocr_box="", ocr_color=""):
-    print("image_array: ", image_array)
-    print(got_mode, ' ', ocr_box, ' ', ocr_color)
-    # image = Image.fromarray(np.uint8(image_array))
     image = image_array
     if got_mode == "plain texts OCR":
-        res = model.chat(tokenizer, image, ocr_type='ocr', stream_flag = True)
+        res = model.chat(tokenizer, image, ocr_type='ocr')
     elif got_mode == "format texts OCR":
         res = model.chat(tokenizer, image, ocr_type='format', render=True, save_render_file='./demo.html')
     elif got_mode == "plain multi-crop OCR":
@@ -31,8 +28,8 @@ def run_GOT(image_array, got_mode, ocr_box="", ocr_color=""):
     if "format" in got_mode:
         with open('./demo.html', 'r') as f:
             demo_html = f.read()
-        yield res, demo_html
-    yield res, None
+        return res, demo_html
+    return res, None
 
 def task_update(task):
     if "fine-grained" in task:
@@ -106,8 +103,7 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Submit")
         
         with gr.Column():
-            chatbot = gr.Chatbot(label="GOT output")
-            # ocr_result = gr.Textbox(label="GOT output")
+            ocr_result = gr.Textbox(label="GOT output")
             html_result = gr.HTML(label="rendered html")
     
     gr.Examples(
@@ -121,7 +117,8 @@ with gr.Blocks() as demo:
             ["assets/color.png", "plain fine-grained OCR", "color", "red", ""],
         ],
         inputs=[image_input, task_dropdown, fine_grained_dropdown, color_dropdown, box_input],
-        outputs=[chatbot, html_result],
+        outputs=[ocr_result, html_result],
+        fn = run_GOT,
         label="examples",
     )
 
@@ -139,7 +136,7 @@ with gr.Blocks() as demo:
     submit_button.click(
         run_GOT,
         inputs=[image_input, task_dropdown, box_input, color_dropdown],
-        outputs=[chatbot, html_result]
+        outputs=[ocr_result, html_result]
     )
 
 demo.launch(share=True)