Spaces:
Runtime error
Runtime error
return formatted res
Browse files
app.py
CHANGED
@@ -15,6 +15,7 @@ import tempfile
|
|
15 |
import time
|
16 |
import shutil
|
17 |
import cv2
|
|
|
18 |
|
19 |
|
20 |
model_name = 'ucaslcl/GOT-OCR2_0'
|
@@ -138,15 +139,17 @@ def ocr_demo(image, task, ocr_type, ocr_box, ocr_color):
|
|
138 |
return res, None
|
139 |
|
140 |
res = res.replace("\\title", "\\title ")
|
141 |
-
|
|
|
|
|
142 |
|
143 |
if html_content:
|
144 |
encoded_html = base64.b64encode(html_content.encode('utf-8')).decode('utf-8')
|
145 |
iframe_src = f"data:text/html;base64,{encoded_html}"
|
146 |
iframe = f'<iframe src="{iframe_src}" width="100%" height="600px"></iframe>'
|
147 |
download_link = f'<a href="data:text/html;base64,{encoded_html}" download="result_{unique_id}.html">Download Full Result</a>'
|
148 |
-
return
|
149 |
-
return
|
150 |
|
151 |
def cleanup_old_files():
|
152 |
current_time = time.time()
|
|
|
15 |
import time
|
16 |
import shutil
|
17 |
import cv2
|
18 |
+
import re
|
19 |
|
20 |
|
21 |
model_name = 'ucaslcl/GOT-OCR2_0'
|
|
|
139 |
return res, None
|
140 |
|
141 |
res = res.replace("\\title", "\\title ")
|
142 |
+
lines = re.split(r'\\\\', res) # Split on double backslashes
|
143 |
+
formatted_lines = [f"$$ {line.strip()} $$" for line in lines if line.strip()]
|
144 |
+
formatted_res = "\n".join(formatted_lines)
|
145 |
|
146 |
if html_content:
|
147 |
encoded_html = base64.b64encode(html_content.encode('utf-8')).decode('utf-8')
|
148 |
iframe_src = f"data:text/html;base64,{encoded_html}"
|
149 |
iframe = f'<iframe src="{iframe_src}" width="100%" height="600px"></iframe>'
|
150 |
download_link = f'<a href="data:text/html;base64,{encoded_html}" download="result_{unique_id}.html">Download Full Result</a>'
|
151 |
+
return formatted_res, f"{download_link}<br>{iframe}"
|
152 |
+
return formatted_res, None
|
153 |
|
154 |
def cleanup_old_files():
|
155 |
current_time = time.time()
|
globe.py
CHANGED
@@ -40,6 +40,7 @@ modelinfor = """
|
|
40 |
- **Model Name**: GOT-OCR 2.0
|
41 |
- **Hugging Face Repository**: [ucaslcl/GOT-OCR2_0](https://huggingface.co/ucaslcl/GOT-OCR2_0)
|
42 |
- **Environment**: CUDA 11.8 + PyTorch 2.0.1
|
|
|
43 |
"""
|
44 |
|
45 |
tasks = [
|
|
|
40 |
- **Model Name**: GOT-OCR 2.0
|
41 |
- **Hugging Face Repository**: [ucaslcl/GOT-OCR2_0](https://huggingface.co/ucaslcl/GOT-OCR2_0)
|
42 |
- **Environment**: CUDA 11.8 + PyTorch 2.0.1
|
43 |
+
|
44 |
"""
|
45 |
|
46 |
tasks = [
|