diff --git a/docutranslate/__init__.py b/docutranslate/__init__.py index a1154c2..2716290 100644 --- a/docutranslate/__init__.py +++ b/docutranslate/__init__.py @@ -1 +1 @@ -__version__="1.1.0" \ No newline at end of file +__version__="1.1.1" \ No newline at end of file diff --git a/docutranslate/utils/docling_utils.py b/docutranslate/utils/docling_utils.py index 51a92be..f1d9255 100644 --- a/docutranslate/utils/docling_utils.py +++ b/docutranslate/utils/docling_utils.py @@ -1,5 +1,11 @@ -from docling.pipeline.standard_pdf_pipeline import StandardPdfPipeline - +# from docling.pipeline.standard_pdf_pipeline import StandardPdfPipeline +import docling.utils.model_downloader def get_docling_artifacts(): - path = StandardPdfPipeline.download_models_hf() - return path \ No newline at end of file + # path = StandardPdfPipeline.download_models_hf() + path=docling.utils.model_downloader.download_models() + print(f"docling模型包已经下载到{path.resolve()}") + return path +# +if __name__ == '__main__': + get_docling_artifacts() +