Spaces:
Sleeping
Sleeping
updated name issue
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ class_names = ['dosa', 'idly', 'kichdi', 'pongal', 'poori', 'vada']
|
|
13 |
|
14 |
# Create model
|
15 |
effnetb2, effnetb2_transforms = create_effnetb2_model(
|
16 |
-
num_classes=
|
17 |
)
|
18 |
|
19 |
# Load saved weights
|
|
|
13 |
|
14 |
# Create model
|
15 |
effnetb2, effnetb2_transforms = create_effnetb2_model(
|
16 |
+
num_classes=6, # could also use len(class_names)
|
17 |
)
|
18 |
|
19 |
# Load saved weights
|