Spaces:

pamixsun
/

glaucoma_screening

Runtime error

pamixsun commited on Sep 8, 2023

Commit

053a66b

1 Parent(s): a21a9d8

Update glaucoma.py

Files changed (1) hide show

glaucoma.py CHANGED Viewed

@@ -86,7 +86,7 @@ class GlaucomaModel(object):
             vcdr = simple_vcdr(disc_cup)
         except:
             vcdr = np.nan
-        _, disc_cup_image = add_mask(image, disc_cup, [0, 1, 2], [[0, 0, 0], [0, 255, 0], [255, 0, 0]], 0.2)
         return disease_idx, disc_cup_image, cam, vcdr

             vcdr = simple_vcdr(disc_cup)
         except:
             vcdr = np.nan
+        _, disc_cup_image = add_mask(image, disc_cup, [1, 2], [[0, 255, 0], [255, 0, 0]], 0.2)
         return disease_idx, disc_cup_image, cam, vcdr