Update processing_florence2.py
Browse files- 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)
|