RoAr777 commited on
Commit
9c75eb5
·
verified ·
1 Parent(s): d910f84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = index2.search(query_embedding, k=top_k)
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]))):