Update Model_Seg.py

Model_Seg.py

@@ -40,7 +40,7 @@ model = UNet(
 )
 
 checkpoint_path = 'segmentation_model.pt'
-checkpoint = torch.load(checkpoint_path, map_location='cpu')  
+checkpoint = torch.load(checkpoint_path, map_location='cpu', weights_only=False)  
 assert model.state_dict().keys() == checkpoint['network'].keys(), "Model and checkpoint keys do not match"
 
 model.load_state_dict(checkpoint['network'])