diff --git a/loader/image_loader.py b/loader/image_loader.py index 1013e82..b14899a 100644 --- a/loader/image_loader.py +++ b/loader/image_loader.py @@ -30,7 +30,7 @@ class UnstructuredPaddleImageLoader(UnstructuredFileLoader): if __name__ == "__main__": - filepath = "../content/samples/test.jpg" + filepath = os.path.join(os.path.dirname(os.path.dirname(__file__)), "content", "samples", "test.jpg") loader = UnstructuredPaddleImageLoader(filepath, mode="elements") docs = loader.load() for doc in docs: diff --git a/loader/pdf_loader.py b/loader/pdf_loader.py index a27eec1..cb972a9 100644 --- a/loader/pdf_loader.py +++ b/loader/pdf_loader.py @@ -46,7 +46,7 @@ class UnstructuredPaddlePDFLoader(UnstructuredFileLoader): if __name__ == "__main__": - filepath = "../content/samples/test.pdf" + filepath = os.path.join(os.path.dirname(os.path.dirname(__file__)), "content", "samples", "test.pdf") loader = UnstructuredPaddlePDFLoader(filepath, mode="elements") docs = loader.load() for doc in docs: