Spaces:
Sleeping
Sleeping
Updates model name
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
|
|
20 |
# Load saved weights
|
21 |
effnetb2.load_state_dict(
|
22 |
torch.load(
|
23 |
-
f="
|
24 |
map_location=torch.device("cpu"), # load to CPU
|
25 |
)
|
26 |
)
|
|
|
20 |
# Load saved weights
|
21 |
effnetb2.load_state_dict(
|
22 |
torch.load(
|
23 |
+
f="09_effnetb2_feature_extractor.pth",
|
24 |
map_location=torch.device("cpu"), # load to CPU
|
25 |
)
|
26 |
)
|