Spaces:
Build error
Build error
Commit
·
2e15984
1
Parent(s):
404e8d7
debug
Browse files
app.py
CHANGED
|
@@ -133,7 +133,10 @@ def search(query: str, ds, images, k):
|
|
| 133 |
top_k_indices = scores.argsort(axis=1)[0][-k:][::-1]
|
| 134 |
|
| 135 |
results = []
|
|
|
|
|
|
|
| 136 |
for idx in top_k_indices:
|
|
|
|
| 137 |
results.append((images[idx])) #, f"Page {idx}"
|
| 138 |
print(f"Temps: {time.time()- start} s")
|
| 139 |
del model
|
|
|
|
| 133 |
top_k_indices = scores.argsort(axis=1)[0][-k:][::-1]
|
| 134 |
|
| 135 |
results = []
|
| 136 |
+
print(images)
|
| 137 |
+
print(top_k_indices)
|
| 138 |
for idx in top_k_indices:
|
| 139 |
+
print(idx)
|
| 140 |
results.append((images[idx])) #, f"Page {idx}"
|
| 141 |
print(f"Temps: {time.time()- start} s")
|
| 142 |
del model
|