fix: remove unused variable

luxonis · Dec 11, 2023 · 2757a96 · 2757a96
1 parent ff4ad53
commit 2757a96
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/datadreamer/dataset_annotation/kosmos2_annotator.py b/src/datadreamer/dataset_annotation/kosmos2_annotator.py
@@ -88,9 +88,6 @@ def annotate(self, image, prompts, conf_threshold=0.1, use_tta=False):
         generated_text = self.processor.batch_decode(
             generated_ids, skip_special_tokens=True
         )[0]
-        processed_text = self.processor.post_process_generation(
-            generated_text, cleanup_and_extract=False
-        )
 
         caption, entities = self.processor.post_process_generation(generated_text)