Merge pull request #179 from gkumbhat/fix_offsets

🐛 Fix offset for standalone content results
foundation-model-stack · Aug 23, 2024 · d1c16f4 · d1c16f4
2 parents 52bc871 + 6e0b9f9
commit d1c16f4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/orchestrator/unary.rs b/src/orchestrator/unary.rs
@@ -594,10 +594,14 @@ pub async fn detect_content(
     let results = chunks
         .into_iter()
         .zip(response)
-        .flat_map(|(_chunk, response)| {
+        .flat_map(|(chunk, response)| {
             response
                 .into_iter()
-                .filter_map(|resp| (resp.score >= threshold).then_some(resp))
+                .filter_map(|mut resp| {
+                    resp.start += chunk.offset;
+                    resp.end += chunk.offset;
+                    (resp.score >= threshold).then_some(resp)
+                })
                 .collect::<Vec<_>>()
         })
         .collect::<Vec<_>>();