Spaces:
Sleeping
Sleeping
Update app.py
Browse filescorrected module creation call
app.py
CHANGED
|
@@ -12,7 +12,7 @@ class_names = ['pizza', 'steak', 'sushi']
|
|
| 12 |
|
| 13 |
### 2. Model and transforms setup
|
| 14 |
effnetb2, effnetb2_transforms = create_effnetb2_model(
|
| 15 |
-
|
| 16 |
)
|
| 17 |
|
| 18 |
#Load the saved weights
|
|
|
|
| 12 |
|
| 13 |
### 2. Model and transforms setup
|
| 14 |
effnetb2, effnetb2_transforms = create_effnetb2_model(
|
| 15 |
+
output_classes=3,
|
| 16 |
)
|
| 17 |
|
| 18 |
#Load the saved weights
|