Spaces:
Sleeping
Sleeping
Update models/langOpen.py
Browse files- models/langOpen.py +5 -3
models/langOpen.py
CHANGED
|
@@ -7,6 +7,10 @@ from langchain.chat_models import ChatOpenAI
|
|
| 7 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
| 8 |
from langchain.prompts import PromptTemplate
|
| 9 |
from langchain.vectorstores import FAISS
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
load_dotenv()
|
| 12 |
|
|
@@ -31,9 +35,7 @@ class LangOpen:
|
|
| 31 |
if os.path.exists(path=path):
|
| 32 |
return FAISS.load_local(folder_path=path, embeddings=embeddings)
|
| 33 |
else:
|
| 34 |
-
faiss = FAISS.
|
| 35 |
-
"./assets/updated_calregs.txt", embedding=embeddings
|
| 36 |
-
)
|
| 37 |
faiss.save_local(path)
|
| 38 |
return faiss
|
| 39 |
|
|
|
|
| 7 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
| 8 |
from langchain.prompts import PromptTemplate
|
| 9 |
from langchain.vectorstores import FAISS
|
| 10 |
+
from langchain.document_loaders import PyPDFLoader
|
| 11 |
+
|
| 12 |
+
loader = PyPDFLoader("./assets/CADWReg.pdf")
|
| 13 |
+
pages = loader.load_and_split()
|
| 14 |
|
| 15 |
load_dotenv()
|
| 16 |
|
|
|
|
| 35 |
if os.path.exists(path=path):
|
| 36 |
return FAISS.load_local(folder_path=path, embeddings=embeddings)
|
| 37 |
else:
|
| 38 |
+
faiss = FAISS.from_documents(pages, embedding)
|
|
|
|
|
|
|
| 39 |
faiss.save_local(path)
|
| 40 |
return faiss
|
| 41 |
|