myhloli commited on
Commit
ee6fe06
·
verified ·
1 Parent(s): b5d228a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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 = DiskReaderWriter(os.path.dirname(path))
36
- return disk_rw.read(os.path.basename(path), AbsReaderWriter.MODE_BIN)
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):