Spaces:

SahithiR
/

Assignment12

Runtime error

SahithiR commited on Jul 31, 2023

Commit

3fc77ff

1 Parent(s): 91a1f76

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,5 +9,5 @@ from CUSTOMRESNET import Net
 model=Net()
 model.load_state_dict(torch.load("model.pth",map_location=torch.device('cpu')),strict=False)
- classes = ('plane', 'car', 'bird', 'cat',
            'deer', 'dog', 'frog', 'horse', 'ship', 'truck')

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