Spaces:

SahithiR
/

Assignment12

Runtime error

SahithiR commited on Aug 17, 2023

Commit

7de7094

1 Parent(s): ef3a63f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ from Model.Lit_cifar_module import LitCifar
 from utils import *
 model = LitCifar().cpu()
-model.load_state_dict(torch.load('final_dict.pth', map_location=torch.device('cpu')))
 model.eval()
 classes = ('plane', 'car', 'bird', 'cat', 'deer',

 from utils import *
 model = LitCifar().cpu()
+model.load_state_dict(torch.load('model_weights.pth', map_location=torch.device('cpu')))
 model.eval()
 classes = ('plane', 'car', 'bird', 'cat', 'deer',