Spaces:
Running
Running
added softmax temperature for better uncertainty calibration
Browse files- app/model.py +1 -1
app/model.py
CHANGED
|
@@ -10,7 +10,7 @@ from transformers import AutoImageProcessor, AutoModelForImageClassification
|
|
| 10 |
HF_NAME = "microsoft/swinv2-base-patch4-window16-256"
|
| 11 |
WEIGHTS_PATH = Path(__file__).parent.parent / "model.safetensors"
|
| 12 |
NUM_LABELS = 2
|
| 13 |
-
SOFTMAX_TEMPERATURE =
|
| 14 |
|
| 15 |
_device = torch.device("cpu")
|
| 16 |
_processor = None
|
|
|
|
| 10 |
HF_NAME = "microsoft/swinv2-base-patch4-window16-256"
|
| 11 |
WEIGHTS_PATH = Path(__file__).parent.parent / "model.safetensors"
|
| 12 |
NUM_LABELS = 2
|
| 13 |
+
SOFTMAX_TEMPERATURE = 2.0
|
| 14 |
|
| 15 |
_device = torch.device("cpu")
|
| 16 |
_processor = None
|