Spaces:

dhruv2842
/

glaucoma_backend_2

Sleeping

dhruv2842 commited on Jun 22

Commit

93d636a

verified ·

1 Parent(s): 4ca7330

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,8 @@ model.load_state_dict(torch.load('densenet169_seed40_best.pt', map_location='cpu
 model.eval()
 # ✅ Class Names
-CLASS_NAMES = ["Normal", "Early Glaucoma", "Advanced Glaucoma"]
 # ✅ Transformation for input images
 transform = transforms.Compose([

 model.eval()
 # ✅ Class Names
+class_names = ["Advanced", "Early", "Normal"]
 # ✅ Transformation for input images
 transform = transforms.Compose([