Update app.py

app.py

@@ -87,10 +87,10 @@ def ocr_image_gradio_stream(image, max_tokens=4096):
         for new_text in streamer:
             generated_text += new_text
             processed_text = process_tags(generated_text)
-            yield processed_text, gr.update(interactive=False)
+            yield process_tags(processed_text), gr.update(interactive=False)
         
         # Re-enable button when complete
-        yield processed_text, gr.update(interactive=True)
+        yield process_tags(processed_text), gr.update(interactive=True)
             
     except Exception as e:
         error_msg = f"Error processing image: {str(e)}"
@@ -196,12 +196,12 @@ with gr.Blocks(title="Nanonets OCR Demo") as demo:
         show_progress=True
     )
     
-    image_input.change(
-        fn=ocr_image_gradio_stream,
-        inputs=[image_input, max_tokens_slider],
-        outputs=[output_text, extract_btn],
-        show_progress=True
-    )
+    # image_input.change(
+    #     fn=ocr_image_gradio_stream,
+    #     inputs=[image_input, max_tokens_slider],
+    #     outputs=[output_text, extract_btn],
+    #     show_progress=True
+    # )
     
     # Add model information section
     with gr.Accordion("About Nanonets-OCR-s", open=False):