From 5f0697799a90c9ad1fa3db2c73de2eccf254a2d0 Mon Sep 17 00:00:00 2001 From: ysyx2008 Date: Tue, 11 Jul 2023 19:12:46 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=20image=5Floader.py=20?= =?UTF-8?q?=E8=BF=90=E8=A1=8C=E6=97=B6=E6=8A=A5=E9=94=99=E2=80=9CModuleNot?= =?UTF-8?q?FoundError:=20No=20module=20named=20'configs'=E2=80=9D=E7=9A=84?= =?UTF-8?q?=E9=97=AE=E9=A2=98=E3=80=82=20(#795)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- loader/image_loader.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/loader/image_loader.py b/loader/image_loader.py index ec32459..4ac4c51 100644 --- a/loader/image_loader.py +++ b/loader/image_loader.py @@ -5,9 +5,6 @@ from langchain.document_loaders.unstructured import UnstructuredFileLoader from paddleocr import PaddleOCR import os import nltk -from configs.model_config import NLTK_DATA_PATH - -nltk.data.path = [NLTK_DATA_PATH] + nltk.data.path class UnstructuredPaddleImageLoader(UnstructuredFileLoader): """Loader that uses unstructured to load image files, such as PNGs and JPGs.""" @@ -35,6 +32,10 @@ class UnstructuredPaddleImageLoader(UnstructuredFileLoader): if __name__ == "__main__": import sys sys.path.append(os.path.dirname(os.path.dirname(__file__))) + + from configs.model_config import NLTK_DATA_PATH + nltk.data.path = [NLTK_DATA_PATH] + nltk.data.path + filepath = os.path.join(os.path.dirname(os.path.dirname(__file__)), "knowledge_base", "samples", "content", "test.jpg") loader = UnstructuredPaddleImageLoader(filepath, mode="elements") docs = loader.load()