--- a
+++ b/class_names.py
@@ -0,0 +1,24 @@
+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',
+                     ]