microsoft
/

Florence-2-large

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

Fix runon lines

#16

by milekpl - opened Jun 21

base: refs/heads/main

←

from: refs/pr/16

Discussion Files changed

Files changed (1) hide show

processing_florence2.py +1 -1

processing_florence2.py CHANGED Viewed

@@ -324,7 +324,7 @@ class Florence2Processor(ProcessorMixin):
         if task_answer_post_processing_type == 'pure_text':
             final_answer = task_answer
             # remove the special tokens
-            final_answer = final_answer.replace('<s>', '').replace('</s>', '')
         elif task_answer_post_processing_type in ['od', 'description_with_bboxes', 'bboxes']:
             od_instances = task_answer
             bboxes_od = [_od_instance['bbox'] for _od_instance in od_instances]

         if task_answer_post_processing_type == 'pure_text':
             final_answer = task_answer
             # remove the special tokens
+            final_answer = final_answer.replace('<s>', '').replace('</s>', '\n')
         elif task_answer_post_processing_type in ['od', 'description_with_bboxes', 'bboxes']:
             od_instances = task_answer
             bboxes_od = [_od_instance['bbox'] for _od_instance in od_instances]