Adrien commited on
Commit
c56c0fd
·
1 Parent(s): 9cb6543
Files changed (1) hide show
  1. rag_demo/rag/source_annotator.py +1 -1
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.lower(), chunk.content.lower())
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)