Spaces:

Cherryblade29
/

FoodVision3

Sleeping

Cherryblade29 commited on Jul 21, 2024

Commit

c89862f

verified ·

1 Parent(s): 112a3a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,9 +16,9 @@ effnetb2 , effnetb2_transforms = create_effnetb2_model(num_classes=len(class_nam
 # load and save weights
-#effnetb2.load_state_dict(torch.load(os.path.join("effnetb2.pth"),map_location=torch.device('cpu')))
-effnetb2.load_state_dict(torch.load("effnetb2.pth",map_location=torch.device('cpu')))
 # Predict function

 # load and save weights
+effnetb2.load_state_dict(torch.load(os.path.join("effnetb2.pth"),map_location=torch.device('cpu')))
+#effnetb2.load_state_dict(torch.load("effnetb2.pth",map_location=torch.device('cpu')))
 # Predict function