Spaces:
Running
on
L40S
Running
on
L40S
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):
|