Update retrival.py

retrival.py

@@ -18,6 +18,7 @@ pytesseract.pytesseract.tesseract_cmd = (r'/usr/bin/tesseract')
 # Configurations
 UPLOAD_FOLDER = "./uploads"
 VECTOR_DB_FOLDER = "./VectorDB"
+IMAGE_DB_FOLDER = "./Images"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(VECTOR_DB_FOLDER, exist_ok=True)
 
@@ -41,7 +42,7 @@ def load_document(data_path):
             try:
                 # Determine the file type based on extension
                 filename, file_extension = os.path.splitext(file.lower())
-                image_output = f"H:/DEV PATEL/2025/RAG Project/Images/{filename}/"
+                image_output = f"./Images/{filename}/"
                 # Use specific partition techniques based on file extension
                 if file_extension == ".pdf":
                     elements = partition_pdf(