Spaces:
Runtime error
Runtime error
Adrien
commited on
Commit
·
c56c0fd
1
Parent(s):
9cb6543
fix
Browse files
rag_demo/rag/source_annotator.py
CHANGED
|
@@ -23,7 +23,7 @@ class SourceAnnotator:
|
|
| 23 |
for sentence in sentences:
|
| 24 |
scores = []
|
| 25 |
for chunk in reranked_chunks:
|
| 26 |
-
score = self.source_annotator(sentence
|
| 27 |
score["filename"] = chunk.metadata["filename"].split(".pdf")[0]
|
| 28 |
score["chunk_id"] = chunk.chunk_id
|
| 29 |
scores.append(score)
|
|
|
|
| 23 |
for sentence in sentences:
|
| 24 |
scores = []
|
| 25 |
for chunk in reranked_chunks:
|
| 26 |
+
score = self.source_annotator(sentence, chunk.content)
|
| 27 |
score["filename"] = chunk.metadata["filename"].split(".pdf")[0]
|
| 28 |
score["chunk_id"] = chunk.chunk_id
|
| 29 |
scores.append(score)
|