Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -111,6 +111,7 @@ class NearestNeighbors:
|
|
| 111 |
rerank_search = NearestNeighbors(n_neighbors=self.n_neighbors, metric='cosine')
|
| 112 |
rerank_search.fit(rerank_data)
|
| 113 |
re_sim, re_idx = rerank_search.kneighbors(np.asarray([q]))
|
|
|
|
| 114 |
re_sims[i, :] = re_sim
|
| 115 |
re_idxs[i, :] = idx[i][re_idx]
|
| 116 |
idx = re_idxs
|
|
|
|
| 111 |
rerank_search = NearestNeighbors(n_neighbors=self.n_neighbors, metric='cosine')
|
| 112 |
rerank_search.fit(rerank_data)
|
| 113 |
re_sim, re_idx = rerank_search.kneighbors(np.asarray([q]))
|
| 114 |
+
print("re_idx: ", re_idx)
|
| 115 |
re_sims[i, :] = re_sim
|
| 116 |
re_idxs[i, :] = idx[i][re_idx]
|
| 117 |
idx = re_idxs
|