Update app.py

app.py

@@ -80,6 +80,9 @@ class LitResnet(LightningModule):
 
 inference_model = LitResnet.load_from_checkpoint("cifar10_customresnet_20_epoch.ckpt")
 
+classes = ('plane', 'car', 'bird', 'cat', 'deer',
+           'dog', 'frog', 'horse', 'ship', 'truck')
+
 def inference(input_img, see_misclassified,num_misclassified_imgs,see_gradcam,num_gradcam_imgs,transparency = 0.85, target_layer_number = -1,top_classes=3):
 
     # if see_misclassified: # show misclassified images