Update Model_Class.py

set torch.load(weights_only=False) to ensure compatibility with torch > 2.6.

Model_Class.py

@@ -59,7 +59,7 @@ val_transforms_416x628 = Compose(
     ]
 )
 
-checkpoint = torch.load("classification_model.ckpt", map_location=torch.device('cpu'))
+checkpoint = torch.load("classification_model.ckpt", map_location=torch.device('cpu'), weights_only=False)
 model = ResNet()
 model.load_state_dict(checkpoint["state_dict"])
 model.eval()