Spaces:
Running
on
L4
Running
on
L4
Update app.py
Browse files
app.py
CHANGED
@@ -25,15 +25,14 @@ from gradio_pdf import PDF
|
|
25 |
import gradio as gr
|
26 |
from loguru import logger
|
27 |
|
|
|
28 |
from magic_pdf.libs.hash_utils import compute_sha256
|
29 |
-
from magic_pdf.rw.AbsReaderWriter import AbsReaderWriter
|
30 |
-
from magic_pdf.rw.DiskReaderWriter import DiskReaderWriter
|
31 |
from magic_pdf.tools.common import do_parse, prepare_env
|
32 |
|
33 |
|
34 |
def read_fn(path):
|
35 |
-
disk_rw =
|
36 |
-
return disk_rw.read(os.path.basename(path)
|
37 |
|
38 |
|
39 |
def parse_pdf(doc_path, output_dir, end_page_id, is_ocr, layout_mode, formula_enable, table_enable, language):
|
|
|
25 |
import gradio as gr
|
26 |
from loguru import logger
|
27 |
|
28 |
+
from magic_pdf.data.data_reader_writer import FileBasedDataReader
|
29 |
from magic_pdf.libs.hash_utils import compute_sha256
|
|
|
|
|
30 |
from magic_pdf.tools.common import do_parse, prepare_env
|
31 |
|
32 |
|
33 |
def read_fn(path):
|
34 |
+
disk_rw = FileBasedDataReader(os.path.dirname(path))
|
35 |
+
return disk_rw.read(os.path.basename(path))
|
36 |
|
37 |
|
38 |
def parse_pdf(doc_path, output_dir, end_page_id, is_ocr, layout_mode, formula_enable, table_enable, language):
|