cfoli commited on
Commit
b90f476
·
1 Parent(s): c3e34f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ labels_dict = {
113
  """### Create torch lightning model (i.e., classifier) module"""
114
 
115
  class modelModule(torch_light.LightningModule):
116
- def __init__(self, num_classes = configs['NUM_CLASSES'], backbone_model_name = configs['DEFAULT_BACKBONE']):
117
  super().__init__()
118
  self.num_classes = num_classes
119
  self.backbone_model_name = backbone_model_name
 
113
  """### Create torch lightning model (i.e., classifier) module"""
114
 
115
  class modelModule(torch_light.LightningModule):
116
+ def __init__(self, num_classes = configs['NUM_CLASSES'], backbone_model_name = 'efficientnet_b3'):
117
  super().__init__()
118
  self.num_classes = num_classes
119
  self.backbone_model_name = backbone_model_name