Merge pull request open-webui#1050 from jannikstdl/rag-pdf-ocr

feat: added ocr functionality to the pdf loader
sime2408 · Mar 6, 2024 · 8fb5f54 · 8fb5f54
2 parents 2111398 + 089a63e
commit 8fb5f54
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/backend/apps/rag/main.py b/backend/apps/rag/main.py
@@ -425,7 +425,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":

diff --git a/backend/requirements.txt b/backend/requirements.txt
@@ -34,6 +34,7 @@ pandas
 openpyxl
 pyxlsb
 xlrd
+rapidocr-onnxruntime
 
 faster-whisper
-Original file line number
+Diff line change
@@ Expand Up / @@ -34,6 +34,7 @@ pandas @@
     openpyxl
     pyxlsb
     xlrd
+    rapidocr-onnxruntime
     faster-whisper
@@ Expand Down @@