Update app.py

app.py

@@ -142,7 +142,7 @@ labels_dict = {
 """### Create torch lightning model (i.e., classifier) module"""
 
 class modelModule(torch_light.LightningModule):
-    def __init__(self, backbone_model_name, num_layers_to_unfreeze, configs['NUM_CLASSES']):
+    def __init__(self, backbone_model_name, num_layers_to_unfreeze, num_classes=configs['NUM_CLASSES']):
         super().__init__()
         self.num_classes = num_classes
         self.backbone_model_name = backbone_model_name