Spaces:
Sleeping
Sleeping
fix declare wrong variable name effnetb2_ransforms to effnetb2_transforms
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from typing import Tuple, Dict
|
|
11 |
class_names = ['pizza', 'steak', 'sushi']
|
12 |
|
13 |
### 2. MOdel and transforms perparation ###
|
14 |
-
effnetb2,
|
15 |
num_classes=3)
|
16 |
|
17 |
# Load save weights
|
|
|
11 |
class_names = ['pizza', 'steak', 'sushi']
|
12 |
|
13 |
### 2. MOdel and transforms perparation ###
|
14 |
+
effnetb2, effnetb2_transforms = create_effnetb2_model(
|
15 |
num_classes=3)
|
16 |
|
17 |
# Load save weights
|