Commit
•
aec3bc5
1
Parent(s):
8651614
Fix runon lines (#16)
Browse files- Fix runon lines (e83cf618af698c7037d277e6f595ffe536d7ea17)
Co-authored-by: Marcin Miłkowski <[email protected]>
- processing_florence2.py +1 -1
processing_florence2.py
CHANGED
@@ -324,7 +324,7 @@ class Florence2Processor(ProcessorMixin):
|
|
324 |
if task_answer_post_processing_type == 'pure_text':
|
325 |
final_answer = task_answer
|
326 |
# remove the special tokens
|
327 |
-
final_answer = final_answer.replace('<s>', '').replace('</s>', '')
|
328 |
elif task_answer_post_processing_type in ['od', 'description_with_bboxes', 'bboxes']:
|
329 |
od_instances = task_answer
|
330 |
bboxes_od = [_od_instance['bbox'] for _od_instance in od_instances]
|
|
|
324 |
if task_answer_post_processing_type == 'pure_text':
|
325 |
final_answer = task_answer
|
326 |
# remove the special tokens
|
327 |
+
final_answer = final_answer.replace('<s>', '').replace('</s>', '\n')
|
328 |
elif task_answer_post_processing_type in ['od', 'description_with_bboxes', 'bboxes']:
|
329 |
od_instances = task_answer
|
330 |
bboxes_od = [_od_instance['bbox'] for _od_instance in od_instances]
|