Spaces:
Runtime error
Runtime error
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
|