소스 검색

fix: use concrete class instead of abstract class

icecraft 1 년 전
부모
커밋
fa3c453c93
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      projects/gradio_app/app.py

+ 2 - 2
projects/gradio_app/app.py

@@ -13,13 +13,13 @@ import pymupdf
 from gradio_pdf import PDF
 from loguru import logger
 
-from magic_pdf.data.data_reader_writer import DataReader
+from magic_pdf.data.data_reader_writer import FileBasedDataReader
 from magic_pdf.libs.hash_utils import compute_sha256
 from magic_pdf.tools.common import do_parse, prepare_env
 
 
 def read_fn(path):
-    disk_rw = DataReader(os.path.dirname(path))
+    disk_rw = FileBasedDataReader(os.path.dirname(path))
     return disk_rw.read(os.path.basename(path))