DineshKumar1329 commited on
Commit
33c2832
·
verified ·
1 Parent(s): da5c6e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -46,7 +46,7 @@ from torchvision import transforms
46
  from PIL import Image
47
 
48
  # Load the model
49
- model = torch.hub.load('your-username/your-repository', 'resnet_cat_dog_classifier')
50
 
51
  # Define the transformation
52
  transform = transforms.Compose([
 
46
  from PIL import Image
47
 
48
  # Load the model
49
+ model = torch.hub.load('DineshKumar1329/DogCat_Classifier', 'resnet_cat_dog_classifier')
50
 
51
  # Define the transformation
52
  transform = transforms.Compose([