Spaces:

Gosula
/

ERA_S12

Sleeping

Gosula commited on Aug 14, 2023

Commit

c5d735a

•

1 Parent(s): 93e5b84

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,8 +14,6 @@ model = CustomResNet()
 model.load_state_dict(torch.load('CustomResNet.pth', map_location=torch.device('cpu')), strict=False)
 model.eval()
 classes = ('plane', 'car', 'bird', 'cat', 'deer',
             'dog', 'frog', 'horse', 'ship', 'truck')

 model.load_state_dict(torch.load('CustomResNet.pth', map_location=torch.device('cpu')), strict=False)
 model.eval()
 classes = ('plane', 'car', 'bird', 'cat', 'deer',
             'dog', 'frog', 'horse', 'ship', 'truck')