Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model()
|
|
18 |
|
19 |
# Load save weights
|
20 |
effnetb2.load_state_dict(
|
21 |
-
torch.load(
|
22 |
map_location = torch.device("cpu"))) # Load the model to the CPU))
|
23 |
|
24 |
### 3. Predict function ###
|
|
|
18 |
|
19 |
# Load save weights
|
20 |
effnetb2.load_state_dict(
|
21 |
+
torch.load(f = "09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
|
22 |
map_location = torch.device("cpu"))) # Load the model to the CPU))
|
23 |
|
24 |
### 3. Predict function ###
|