Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ pytesseract.pytesseract.tesseract_cmd = r"tesseract.exe"
|
|
13 |
|
14 |
model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
|
15 |
index = faiss.read_index('IPC_index.faiss')
|
16 |
-
index2 = faiss.read_index('CrpC_index.faiss')
|
17 |
|
18 |
|
19 |
# Step 3: Retrieval with Citations using PDF filename
|
@@ -38,7 +38,7 @@ def retrieve_info_with_citation(query, top_k=5):
|
|
38 |
|
39 |
def retrieve_info_with_citation2(query, top_k=5):
|
40 |
query_embedding = model.encode([query])
|
41 |
-
D, I =
|
42 |
|
43 |
results = []
|
44 |
for i in range(min(top_k, len(I[0]))):
|
|
|
13 |
|
14 |
model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
|
15 |
index = faiss.read_index('IPC_index.faiss')
|
16 |
+
#index2 = faiss.read_index('CrpC_index.faiss')
|
17 |
|
18 |
|
19 |
# Step 3: Retrieval with Citations using PDF filename
|
|
|
38 |
|
39 |
def retrieve_info_with_citation2(query, top_k=5):
|
40 |
query_embedding = model.encode([query])
|
41 |
+
D, I = index.search(query_embedding, k=top_k)
|
42 |
|
43 |
results = []
|
44 |
for i in range(min(top_k, len(I[0]))):
|