Spaces:

pamixsun
/

glaucoma_screening

Runtime error

pamixsun commited on Aug 8, 2023

Commit

8cd7940

1 Parent(s): ebe7097

Update glaucoma.py

Files changed (1) hide show

glaucoma.py CHANGED Viewed

@@ -21,7 +21,7 @@ class GlaucomaModel(object):
         # classification id to label
         self.id2label = self.cls_model.config.id2label
-        # number of classes for nails disease
         self.num_diseases = len(self.id2label)
     def glaucoma_pred(self, image):

         # classification id to label
         self.id2label = self.cls_model.config.id2label
+        # number of classes
         self.num_diseases = len(self.id2label)
     def glaucoma_pred(self, image):