Spaces:
Running
on
Zero
Running
on
Zero
Liam Dyer
commited on
Commit
•
b5396be
1
Parent(s):
bab324c
yer
Browse files
app.py
CHANGED
|
@@ -163,8 +163,6 @@ def predict(queries, documents, max_characters) -> list[list[str]]:
|
|
| 163 |
|
| 164 |
# Grab the top score for the query
|
| 165 |
doc_idx, chunk_idx, _ = doc_scores.pop(0)
|
| 166 |
-
if doc_idx not in document_embeddings:
|
| 167 |
-
document_embeddings[doc_idx] = []
|
| 168 |
|
| 169 |
# Ensure we have space
|
| 170 |
chunk = chunked_docs[doc_idx][chunk_idx]
|
|
|
|
| 163 |
|
| 164 |
# Grab the top score for the query
|
| 165 |
doc_idx, chunk_idx, _ = doc_scores.pop(0)
|
|
|
|
|
|
|
| 166 |
|
| 167 |
# Ensure we have space
|
| 168 |
chunk = chunked_docs[doc_idx][chunk_idx]
|