Spaces:

BigTimeCoderSean
/

Car_classifier

Runtime error

BigTimeCoderSean commited on Jan 5, 2023

Commit

25c0ad4

1 Parent(s): 870a28d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ effnetb0.classifier = nn.Sequential(
 )
 #load saved weights
-effnetb0.load_state_dict(torch.load('pretrained_effnetb0_stanford_cars_20_percent.pth'),
                           map_location=torch.device("cpu"))

 )
 #load saved weights
+effnetb0.load_state_dict(torch.load('pretrained_effnetb0_stanford_cars_20_percent.pth',
                           map_location=torch.device("cpu"))