Update glaucoma.py
Browse files- 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
|
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
|