feat: added ocr functionality to the pdf loader

089a63e0 · Jannik Streidl · eb51ad14 · 089a63e0 · 089a63e0
Commit 089a63e0 authored Mar 05, 2024 by Jannik Streidl
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

backend/apps/rag/main.py backend/apps/rag/main.py +1 -1

backend/requirements.txt backend/requirements.txt +1 -0

No files found.
--- a/backend/apps/rag/main.py
+++ b/backend/apps/rag/main.py
@@ -419,7 +419,7 @@ def get_loader(filename: str, file_content_type: str, file_path: str):
    ]
    if file_ext == "pdf":
-        loader = PyPDFLoader(file_path)
+        loader = PyPDFLoader(file_path, extract_images=True)
    elif file_ext == "csv":
        loader = CSVLoader(file_path)
    elif file_ext == "rst":

--- a/backend/requirements.txt
+++ b/backend/requirements.txt
@@ -33,6 +33,7 @@ pandas
 openpyxl
 pyxlsb
 xlrd
+rapidocr-onnxruntime
 faster-whisper