Update app.py

app.py

@@ -11,23 +11,24 @@ model = model.eval().cuda()
 @spaces.GPU
 def run_GOT(image_array, got_mode, fine_grained_mode="", ocr_color="", ocr_box=""):
     image = image_array
+    html_file = './demo.html'
     if got_mode == "plain texts OCR":
         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')
+        res = model.chat(tokenizer, image, ocr_type='format', render=True, save_render_file=html_file)
     elif got_mode == "plain multi-crop OCR":
         res = model.chat_crop(tokenizer, image, ocr_type='ocr')
     elif got_mode == "format multi-crop OCR":
-        res = model.chat_crop(tokenizer, image, ocr_type='format', render=True, save_render_file='./demo.html')
-
+        res = model.chat_crop(tokenizer, image, ocr_type='format', render=True, save_render_file=html_file)
     elif got_mode == "plain fine-grained OCR":
         res = model.chat(tokenizer, image, ocr_type='ocr', ocr_box=ocr_box, ocr_color=ocr_color)
     elif got_mode == "format fine-grained OCR":
-        res = model.chat(tokenizer, image, ocr_type='format', ocr_box=ocr_box, ocr_color=ocr_color, render=True, save_render_file='./demo.html')
+        res = model.chat(tokenizer, image, ocr_type='format', ocr_box=ocr_box, ocr_color=ocr_color, render=True, save_render_file=html_file)
 
     if "format" in got_mode:
-        with open('./demo.html', 'r') as f:
+        with open(html_file, 'r') as f:
             demo_html = f.read()
+        print("demo_html: \n", demo_html)
         return res, demo_html
     return res, None