prediction_classs = ['barretts',

                     'barretts-short-segment',

                     'bbps-0-1',

                     'bbps-2-3'

                     'cecum',

                     'dyed-lifted-polyps',

                     'dyed-resection-margins',

                     'esophagitis-a',

                     'esophagitis-b-d',

                     'hemorrhoids',

                     'ileum',

                     'impacted-stool',

                     'polyps',

                     'pylorus',

                     'retroflex-rectum',

                     'retroflex-stomach',

                     'ulcerative-colitis-grade-0-1',

                     'ulcerative-colitis-grade-1',

                     'ulcerative-colitis-grade-1-2',

                     'ulcerative-colitis-grade-2',

                     'ulcerative-colitis-grade-2-3',

                     'ulcerative-colitis-grade-3',

                     'z-line',

                     ]