Hugging Face's logo

Spaces:
bhardwaj08sarthak
/
demo
Sleeping

App Files Community
bhardwaj08sarthak commited on
Commit
7648e7b
·
verified ·
1 Parent(s): 6791f33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ model= models.resnet18(pretrained=True)
9
  model.fc = torch.nn.Linear(model.fc.in_features,2)
10
  state_dict=torch.load('up500Model.pt', map_location='cpu')
11
  model.load_state_dict(state_dict)
12
- model.eval
13
 
14
  #predictions
15
  imgTransforms = transforms.Comporse([transforms.Resize(256), transforms.CenterCrop(224), trasforms.ToTensor(),transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])])
 
9
  model.fc = torch.nn.Linear(model.fc.in_features,2)
10
  state_dict=torch.load('up500Model.pt', map_location='cpu')
11
  model.load_state_dict(state_dict)
12
+ model.eval()
13
 
14
  #predictions
15
  imgTransforms = transforms.Comporse([transforms.Resize(256), transforms.CenterCrop(224), trasforms.ToTensor(),transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])])