thery.ai / src /utils /main.py
Testys's picture
Using a current method
5889992
raw
history blame contribute delete
277 Bytes
import src.utils.vector_db as vector_db
import src.utils.pdf_splitter as pdf_splitter
vector_db = vector_db.VectorDatabase("mental_health_db")
pdf_process = pdf_splitter.PDFProcessor("../data/mental_health")
processed_pdf = pdf_process.run()
vector_db.create_db(processed_pdf)