darpanaswal commited on
Commit
1f12a21
·
verified ·
1 Parent(s): 1a03059

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ with urllib.request.urlopen(url) as f:
33
  cifar10_classes = {int(k): v[1] for k, v in imagenet_classes.items()}
34
 
35
  # Load the pre-trained ResNet model
36
- model = models.resnet152(pretrained=True)
37
 
38
  # Modify the classifier for 6 classes with an additional hidden layer
39
  # model.fc = nn.Sequential(
 
33
  cifar10_classes = {int(k): v[1] for k, v in imagenet_classes.items()}
34
 
35
  # Load the pre-trained ResNet model
36
+ model = models.resnet18(pretrained=True)
37
 
38
  # Modify the classifier for 6 classes with an additional hidden layer
39
  # model.fc = nn.Sequential(