Update README.md

README.md

@@ -22,7 +22,7 @@ It is essentially an enhancement to cross-entropy loss and is useful for classif
 This repo supports importing modules through `torch.hub`. `FocalLoss` can be easily imported into your code via, for example:
 ```python3
 focal_loss = torch.hub.load(
-	github='adeelh/pytorch-multi-class-focal-loss',
+	repo_or_dir='adeelh/pytorch-multi-class-focal-loss',
 	model='FocalLoss',
 	alpha=torch.tensor([.75, .25]),
 	gamma=2,
@@ -35,7 +35,7 @@ loss = focal_loss(x, y)
 Or:
 ```python3
 focal_loss = torch.hub.load(
-	github='adeelh/pytorch-multi-class-focal-loss',
+	repo_or_dir='adeelh/pytorch-multi-class-focal-loss',
 	model='focal_loss',
 	alpha=[.75, .25],
 	gamma=2,
@@ -46,4 +46,4 @@ focal_loss = torch.hub.load(
 )
 x, y = torch.randn(10, 2), (torch.rand(10) > .5).long()
 loss = focal_loss(x, y)
-```
+```