Update app.py

app.py

@@ -7,11 +7,11 @@ import numpy as np
 tokenizer = AutoTokenizer.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True)
 model = AutoModel.from_pretrained('ucaslcl/GOT-OCR2_0', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True)
 model = model.eval().cuda()
+html_file = './demo.html'
 
 @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":
@@ -128,6 +128,8 @@ with gr.Blocks() as demo:
           \\]
 """
     )
+    with gr.Column():
+        html_show = gr.HTML(f'<a href="{html_file}" target="_blank">Open Demo HTML</a>')
     with gr.Column():
         html_result = gr.HTML(
             value="""