add ocr task (SEACrowd#555)

R-Damanhuri · Apr 1, 2024 · 19c6621 · 19c6621
1 parent 94bc96a
commit 19c6621
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/seacrowd/utils/constants.py b/seacrowd/utils/constants.py
@@ -125,6 +125,7 @@ class Tasks(Enum):
     IMAGE_CAPTIONING = "IC"
     VISUAL_QUESTION_ANSWERING = "VQA"
     SIGN_LANGUAGE_RECOGNITION = "SLR"
+    OPTICAL_CHARACTER_RECOGNITION = "OCR"
     STYLIZED_IMAGE_CAPTIONING = "SIC"
     VISUALLY_GROUNDED_REASONING = "VGR"
 
@@ -282,6 +283,7 @@ class Licenses(Enum):
     Tasks.VISUAL_QUESTION_ANSWERING: "IMQA",
     Tasks.IMAGE_CAPTIONING: "IMTEXT",
     Tasks.SIGN_LANGUAGE_RECOGNITION: "IMTEXT",
+    Tasks.OPTICAL_CHARACTER_RECOGNITION: "IMTEXT",
     Tasks.STYLIZED_IMAGE_CAPTIONING: "IMTEXT",
     Tasks.VISUALLY_GROUNDED_REASONING: "IMTEXT",
     Tasks.VIDEO_CAPTIONING: "VIDTEXT",