using temp file to load pdf

swiss-ai-center · Jan 8, 2024 · 370d331 · 370d331
1 parent ecda4c3
commit 370d331
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -89,7 +89,11 @@ def process(self, data):
             chunk_size=500,
             chunk_overlap=100,
         )
-        loader = PyMuPDFLoader(raw)
+
+        # raw content as PDF temp file
+        with open("temp.pdf", "wb") as f:
+            f.write(raw)
+        loader = PyMuPDFLoader("temp.pdf")
         doc = loader.load_and_split(text_splitter)
         vectorstore = FAISS.from_documents(
             documents=doc,