Spaces:

SahithiR
/

Assignment12

Runtime error

SahithiR commited on Jul 31, 2023

Commit

d690f9b

1 Parent(s): 414fe50

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,3 +9,5 @@ from CUSTOMRESNET import Net
 model=Net()
 model.load_state_dict(torch.load("model.pth",map_location=torch.device('cpu')),strict=False)

 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')