delphiclinic commited on
Commit
f34410c
1 Parent(s): 3b38d62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -11
app.py CHANGED
@@ -12,7 +12,7 @@ from langchain.chains import ConversationalRetrievalChain
12
  from langchain_community.embeddings import HuggingFaceBgeEmbeddings
13
  from langchain_community.vectorstores import FAISS
14
  from langchain_community.document_loaders import PyPDFLoader
15
- from dotenv import load_dotenv, find_dotenv
16
  import pickle
17
  import os
18
  from langchain_community.document_loaders import PyPDFDirectoryLoader
@@ -21,17 +21,22 @@ from langchain_community.document_loaders import PyPDFDirectoryLoader
21
 
22
  huggingfacehub_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
23
 
24
- # loader = UnstructuredPowerPointLoader("data/Hypertension.ppt")
25
 
26
- # documents = loader.load_and_split()
27
 
 
 
 
 
 
 
 
 
 
 
 
28
 
29
- # loader = PyPDFDirectoryLoader("wolo/")
30
-
31
- # documents = loader.load_and_split()
32
-
33
  # Define model and vector store
34
-
35
  embeddings = "BAAI/bge-base-en"
36
  encode_kwargs = {'normalize_embeddings': True}
37
  model_norm = HuggingFaceBgeEmbeddings(
@@ -45,9 +50,7 @@ model_norm = HuggingFaceBgeEmbeddings(
45
  # # saving the embeddings locally
46
  # vector_store.save_local("wolo_database")
47
 
48
- ##loading
49
- vector_store = FAISS.load_local("wolo_database", model_norm, allow_dangerous_deserialization=True)
50
- job_done = object()
51
 
52
 
53
 
 
12
  from langchain_community.embeddings import HuggingFaceBgeEmbeddings
13
  from langchain_community.vectorstores import FAISS
14
  from langchain_community.document_loaders import PyPDFLoader
15
+ # from dotenv import load_dotenv, find_dotenv
16
  import pickle
17
  import os
18
  from langchain_community.document_loaders import PyPDFDirectoryLoader
 
21
 
22
  huggingfacehub_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
23
 
 
24
 
 
25
 
26
+ try:
27
+ ##loading
28
+ vector_store = FAISS.load_local("wolo_database", model_norm, allow_dangerous_deserialization=True)
29
+ job_done = object()
30
+ except:
31
+
32
+ loader = PyPDFDirectoryLoader("wolo/")
33
+
34
+ documents = loader.load_and_split()
35
+
36
+
37
 
38
+
 
 
 
39
  # Define model and vector store
 
40
  embeddings = "BAAI/bge-base-en"
41
  encode_kwargs = {'normalize_embeddings': True}
42
  model_norm = HuggingFaceBgeEmbeddings(
 
50
  # # saving the embeddings locally
51
  # vector_store.save_local("wolo_database")
52
 
53
+
 
 
54
 
55
 
56