Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -26,12 +26,11 @@ def run_GOT(image_array, got_mode, fine_grained_mode="", ocr_color="", ocr_box="
|
|
26 |
elif got_mode == "format fine-grained OCR":
|
27 |
res = model.chat(tokenizer, image, ocr_type='format', ocr_box=ocr_box, ocr_color=ocr_color, render=True, save_render_file=html_file)
|
28 |
|
29 |
-
print("res:\n", res)
|
30 |
if "format" in got_mode:
|
31 |
with open(html_file, 'r') as f:
|
32 |
demo_html = f.read()
|
33 |
-
|
34 |
-
print("demo_html: \n", demo_html)
|
35 |
print(os.path.abspath(html_file))
|
36 |
return res, demo_html
|
37 |
return res, None
|
@@ -76,6 +75,14 @@ with gr.Blocks() as demo:
|
|
76 |
"🔥🔥🔥This is the official online demo of GOT-OCR-2.0 model!!!"
|
77 |
|
78 |
### Demo Guidelines
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
79 |
""")
|
80 |
|
81 |
with gr.Row():
|
@@ -113,8 +120,7 @@ with gr.Blocks() as demo:
|
|
113 |
with gr.Column():
|
114 |
ocr_result = gr.Textbox(label="GOT output")
|
115 |
|
116 |
-
|
117 |
-
html_show = gr.HTML(f'<a href="/home/user/app/demo.html" target="_blank">Open Demo HTML</a>')
|
118 |
with gr.Column():
|
119 |
html_result = gr.HTML(
|
120 |
label="rendered html", show_label=True)
|
|
|
26 |
elif got_mode == "format fine-grained OCR":
|
27 |
res = model.chat(tokenizer, image, ocr_type='format', ocr_box=ocr_box, ocr_color=ocr_color, render=True, save_render_file=html_file)
|
28 |
|
29 |
+
# print("res:\n", res)
|
30 |
if "format" in got_mode:
|
31 |
with open(html_file, 'r') as f:
|
32 |
demo_html = f.read()
|
33 |
+
# print("demo_html: \n", demo_html)
|
|
|
34 |
print(os.path.abspath(html_file))
|
35 |
return res, demo_html
|
36 |
return res, None
|
|
|
75 |
"🔥🔥🔥This is the official online demo of GOT-OCR-2.0 model!!!"
|
76 |
|
77 |
### Demo Guidelines
|
78 |
+
|
79 |
+
You need to upload your image below and choose one mode of GOT, then click "Submit" to run GOT model. More characters will result in longer wait times.
|
80 |
+
- **plain texts OCR & format texts OCR**
|
81 |
+
- The two modes are for the image-level OCR.
|
82 |
+
- **plain multi-crop OCR & format multi-crop OCR**
|
83 |
+
- For images with more complex content, you can achieve higher-quality results with these modes.
|
84 |
+
- **plain fine-grained OCR && format fine-grained OCR**
|
85 |
+
- In these modes, you can specify fine-grained regions on the input image for more flexible OCR. Fine-grained regions can be coordinates of the box, red color, blue color, or green color.
|
86 |
""")
|
87 |
|
88 |
with gr.Row():
|
|
|
120 |
with gr.Column():
|
121 |
ocr_result = gr.Textbox(label="GOT output")
|
122 |
|
123 |
+
|
|
|
124 |
with gr.Column():
|
125 |
html_result = gr.HTML(
|
126 |
label="rendered html", show_label=True)
|