Spaces:
Running
Running
Update app.py
Browse files
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:
|