Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 =
|
| 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
|