diff --git a/loader/image_loader.py b/loader/image_loader.py index 1013e822..b14899ab 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 a27eec12..cb972a9a 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: