HemaAM commited on
Commit
22dd6ba
·
1 Parent(s): 0586402

Update app.py

Browse files

Corrected the resnet model state file name

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('custom_resnet_model.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('cust_resnet_model.pth', map_location=torch.device('cpu')), strict=False)
39
  model1.eval()
40
 
41
  transform = transforms.Compose([