diff --git a/seacrowd/utils/constants.py b/seacrowd/utils/constants.py index 2e8436511..eea6ee1ca 100644 --- 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",