Update app.py

app.py

@@ -23,6 +23,7 @@ from torchvision.models import resnet50
 model = resnet50(weights=None)
 
 # Replace fc with the multi-layer head that matches checkpoint
+num_classes = 196
 in_ch = model.fc.in_features
 model.fc = torch.nn.Linear(in_ch, num_classes)
 # model.fc = torch.nn.Sequential()