silveroxides commited on
Commit
feba260
·
verified ·
1 Parent(s): ce7f104

Update processing_florence2.py

Browse files
Files changed (1) hide show
  1. processing_florence2.py +2 -0
processing_florence2.py CHANGED
@@ -108,6 +108,7 @@ class Florence2Processor(ProcessorMixin):
108
  '<REGION_TO_CATEGORY>': 'pure_text',
109
  '<REGION_TO_DESCRIPTION>': 'pure_text',
110
  '<REGION_TO_OCR>': 'pure_text',
 
111
  '<REGION_PROPOSAL>': 'bboxes'
112
  }
113
 
@@ -130,6 +131,7 @@ class Florence2Processor(ProcessorMixin):
130
  '<REGION_TO_CATEGORY>': 'What is the region {input}?',
131
  '<REGION_TO_DESCRIPTION>': 'What does the region {input} describe?',
132
  '<REGION_TO_OCR>': 'What text is in the region {input}?',
 
133
  }
134
 
135
  self.post_processor = Florence2PostProcesser(tokenizer=tokenizer)
 
108
  '<REGION_TO_CATEGORY>': 'pure_text',
109
  '<REGION_TO_DESCRIPTION>': 'pure_text',
110
  '<REGION_TO_OCR>': 'pure_text',
111
+ '<STYLE1_FURRY_CAPTION>': 'pure_text',
112
  '<REGION_PROPOSAL>': 'bboxes'
113
  }
114
 
 
131
  '<REGION_TO_CATEGORY>': 'What is the region {input}?',
132
  '<REGION_TO_DESCRIPTION>': 'What does the region {input} describe?',
133
  '<REGION_TO_OCR>': 'What text is in the region {input}?',
134
+ '<STYLE1_FURRY_CAPTION>': '{input}STYLE1 FURRY CAPTION:',
135
  }
136
 
137
  self.post_processor = Florence2PostProcesser(tokenizer=tokenizer)