| from backend.ml_models.tornado_predictor import TornadoSuperPredictor as Model | |
| import torch | |
| def load(device='cpu'): | |
| m = Model().to(device) | |
| m.load_state_dict(torch.load('pytorch_model.bin', map_location=device)) | |
| m.eval(); return m | |
| def apply_temperature(logits, T): | |
| return logits / max(T,1e-6) | |