made a minor update to the "Usage" section in README.md

README.md

@@ -38,8 +38,8 @@ class IDENTITY(nn.Module, PyTorchModelHubMixin):
 
     def forward(self, x):
         a = x
-        a = torch.cat([a, a], dim=1)
-        column_sum = torch.sum(a, dim=1, keepdim=True)
+        inputs = torch.cat([a, a], dim=1)
+        column_sum = torch.sum(inputs, dim=1, keepdim=True)
         output = (column_sum >= 2).long()
         return output