Spaces:
Sleeping
Sleeping
shukurullo2004
commited on
Commit
•
058bf4f
1
Parent(s):
c7fc65b
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
|
|
21 |
# Load saved weights
|
22 |
effnetb2.load_state_dict(
|
23 |
torch.load(
|
24 |
-
f="
|
25 |
map_location=torch.device("cpu"), # load to CPU
|
26 |
)
|
27 |
)
|
|
|
21 |
# Load saved weights
|
22 |
effnetb2.load_state_dict(
|
23 |
torch.load(
|
24 |
+
f="food_model.pth",
|
25 |
map_location=torch.device("cpu"), # load to CPU
|
26 |
)
|
27 |
)
|