Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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(
|