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',
]