updated app.py

app.py

@@ -15,7 +15,10 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model_kwargs = {"device": device}
 embeddings = HuggingFaceEmbeddings(model_name=model_name, model_kwargs=model_kwargs)
 
-books_db = Chroma(persist_directory="./chroma_companies", embedding_function=embeddings)
+loader = DirectoryLoader('./companies', glob="**/*.pdf", recursive=True, use_multithreading=True)
+docs = loader.load()
+vectordb = Chroma.from_documents(documents=docs, embedding=embeddings, persist_directory="companies_db")
+books_db = Chroma(persist_directory="./companies_db", embedding_function=embeddings)
 books_db_client = books_db.as_retriever()
 
 # Initialize the model and tokenizer