Penality commited on
Commit
a931303
·
verified ·
1 Parent(s): d2a8511

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,12 +77,12 @@ def retrieve_document(query):
77
  embeddings_file = requests.get(API_URL_EMBEDDINGS)
78
  metadata_file = requests.get(API_URL_METADATA)
79
 
80
- print(embeddings_file.json(), metadata_file.json())
81
  # Generate query embedding
82
  query_embedding = embedding_model.encode([query]).astype(np.float32)
83
 
84
  # Search for the closest document in FAISS index
85
- index = faiss.read_index(embeddings_file)
86
  _, closest_idx = index.search(query_embedding, 1)
87
 
88
  with open(metadata_file, "r") as f:
 
77
  embeddings_file = requests.get(API_URL_EMBEDDINGS)
78
  metadata_file = requests.get(API_URL_METADATA)
79
 
80
+ print(embeddings_file.json()['index_file'], metadata_file.json()['metadata_file'])
81
  # Generate query embedding
82
  query_embedding = embedding_model.encode([query]).astype(np.float32)
83
 
84
  # Search for the closest document in FAISS index
85
+ index = faiss.read_index(embeddings_file.json()['index_file'])
86
  _, closest_idx = index.search(query_embedding, 1)
87
 
88
  with open(metadata_file, "r") as f: