Update Model_Class.py (#3)
Browse files- Update Model_Class.py (51350786c25b786e8236060ca21eb7ac00a180f5)
Co-authored-by: Felix Dorfner <FJDorfner@users.noreply.huggingface.co>
- Model_Class.py +1 -1
Model_Class.py
CHANGED
|
@@ -59,7 +59,7 @@ val_transforms_416x628 = Compose(
|
|
| 59 |
]
|
| 60 |
)
|
| 61 |
|
| 62 |
-
checkpoint = torch.load("classification_model.ckpt", map_location=torch.device('cpu'))
|
| 63 |
model = ResNet()
|
| 64 |
model.load_state_dict(checkpoint["state_dict"])
|
| 65 |
model.eval()
|
|
|
|
| 59 |
]
|
| 60 |
)
|
| 61 |
|
| 62 |
+
checkpoint = torch.load("classification_model.ckpt", map_location=torch.device('cpu'), weights_only=False)
|
| 63 |
model = ResNet()
|
| 64 |
model.load_state_dict(checkpoint["state_dict"])
|
| 65 |
model.eval()
|