pamixsun commited on
Commit
ebe7097
1 Parent(s): 588294a

Update glaucoma.py

Browse files
Files changed (1) hide show
  1. glaucoma.py +1 -1
glaucoma.py CHANGED
@@ -12,7 +12,7 @@ class GlaucomaModel(object):
12
  device=torch.device('cpu')):
13
  # where to load the model, gpu or cpu ?
14
  self.device = device
15
- # classification model for nails disease
16
  self.cls_extractor = AutoImageProcessor.from_pretrained(cls_model_path)
17
  self.cls_model = Swinv2ForImageClassification.from_pretrained(cls_model_path).to(device).eval()
18
  # class activation map
 
12
  device=torch.device('cpu')):
13
  # where to load the model, gpu or cpu ?
14
  self.device = device
15
+ # glaucoma classification model
16
  self.cls_extractor = AutoImageProcessor.from_pretrained(cls_model_path)
17
  self.cls_model = Swinv2ForImageClassification.from_pretrained(cls_model_path).to(device).eval()
18
  # class activation map