Update README.md

README.md

@@ -27,7 +27,7 @@ model.eval()
 
 # dummy input
 batch_size = 1
-H, W, D = 128, 128, 128  # Example dimensions
+H, W, D = 128, 128, 128  
 input_tensor = torch.randn(batch_size, 1, H, W, D)
 
 # Get features