Revert to commit 6c3c1e89b3ab981b79a13afc00dc7e6ddc3f7fe0

app.py

@@ -45,7 +45,7 @@ model = model.eval().cuda()
 model.config.pad_token_id = tokenizer.eos_token_id
 
 @spaces.GPU
-def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, render=True):
+def process_image(image, task, ocr_type=None, ocr_box=None, ocr_color=None, render=False):
     if task == "Plain Text OCR":
         res = model.chat(tokenizer, image, ocr_type='ocr', attention_mask=attention_mask)
     elif task == "Format Text OCR":
@@ -84,8 +84,8 @@ def update_inputs(task):
     elif task == "Render Formatted OCR":
         return [gr.update(visible=False)] * 3 + [gr.update(visible=True)]
     
-def ocr_demo(image, task, ocr_type, ocr_box, ocr_color, render):
-    res, html_content = process_image(image, task, ocr_type, ocr_box, ocr_color, render)
+def ocr_demo(image, task, ocr_type, ocr_box, ocr_color):
+    res, html_content = process_image(image, task, ocr_type, ocr_box, ocr_color)
     if html_content:
         return res, html_content
     return res, None