Update app.py

app.py

@@ -32,7 +32,7 @@ with urllib.request.urlopen(url) as f:
 cifar10_classes = {int(k): v[1] for k, v in imagenet_classes.items()}
 
 # Load the pre-trained ResNet model
-model = models.resnet152(pretrained=True)
+model = models.resnet18(pretrained=True)
 
 # Modify the classifier for 6 classes with an additional hidden layer
 # model.fc = nn.Sequential(