Arthur commited on
Commit
3373f2c
·
1 Parent(s): a30aa26

Fix : OCR_SIGNATURES

Browse files
Files changed (1) hide show
  1. processing_florence2.py +1 -4
processing_florence2.py CHANGED
@@ -120,7 +120,7 @@ class Florence2Processor(ProcessorMixin):
120
  '<OD>': 'Locate the objects with category name in the image.',
121
  '<DENSE_REGION_CAPTION>': 'Locate the objects in the image, with their descriptions.',
122
  '<REGION_PROPOSAL>': 'Locate the region proposals in the image.',
123
- '<OCR_SIGNATURES>': 'Locate the signatures in the image and return the text of the signatures.'
124
  }
125
 
126
  self.task_prompts_with_input = {
@@ -622,9 +622,6 @@ class Florence2PostProcesser(object):
622
  },
623
  {
624
  'TASK_NAME': 'description_with_bboxes_or_polygons',
625
- },
626
- {
627
- 'TASK_NAME': 'ocr_signatures',
628
  }
629
  ]
630
  }
 
120
  '<OD>': 'Locate the objects with category name in the image.',
121
  '<DENSE_REGION_CAPTION>': 'Locate the objects in the image, with their descriptions.',
122
  '<REGION_PROPOSAL>': 'Locate the region proposals in the image.',
123
+ '<OCR_SIGNATURES>': 'Find the signatures in the image if any and return them'
124
  }
125
 
126
  self.task_prompts_with_input = {
 
622
  },
623
  {
624
  'TASK_NAME': 'description_with_bboxes_or_polygons',
 
 
 
625
  }
626
  ]
627
  }