HemaAM commited on
Commit
690c969
·
1 Parent(s): f9f2b4c

Update app.py

Browse files

Fixed the correct model name to load

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ mis_classified_df = mis_classified_df.sample(frac=1)
35
  device = torch.device("cuda")
36
 
37
  model1 = custResNet()
38
- model1.load_state_dict(torch.load('cust_resnet_model_saved1.pth', map_location=torch.device('cpu')), strict=False)
39
  model1.eval()
40
 
41
  transform = transforms.Compose([
 
35
  device = torch.device("cuda")
36
 
37
  model1 = custResNet()
38
+ model1.load_state_dict(torch.load('custom_resnet_model.pth', map_location=torch.device('cpu')), strict=False)
39
  model1.eval()
40
 
41
  transform = transforms.Compose([