Hugging Face's logo

Spaces:
Gosula
/
ERA_S12
Sleeping

App Files Community
4
Gosula commited on
Commit
4394a40
•
1 Parent(s): f51a11d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -9,7 +9,6 @@ from torchvision import datasets, transforms
9
  from custom_resnet import CustomResNet
10
  import random
11
 
12
-
13
  model = CustomResNet()
14
  model.load_state_dict(torch.load('CustomResNet.pth', map_location=torch.device('cpu')), strict=False)
15
  model.eval()
 
9
  from custom_resnet import CustomResNet
10
  import random
11
 
 
12
  model = CustomResNet()
13
  model.load_state_dict(torch.load('CustomResNet.pth', map_location=torch.device('cpu')), strict=False)
14
  model.eval()