Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ class_names = ['pizza', 'steak', 'sushi']
|
|
11 |
|
12 |
|
13 |
#create effnet model
|
14 |
-
effnetb2
|
15 |
num_classes = len(class_names),
|
16 |
seed=42
|
17 |
)
|
|
|
11 |
|
12 |
|
13 |
#create effnet model
|
14 |
+
effnetb2, effnetb2_transforms = create_effnetb2_model(
|
15 |
num_classes = len(class_names),
|
16 |
seed=42
|
17 |
)
|