updated model load code

app.py

@@ -9,6 +9,8 @@ from custom_resnet import Net
 
 # model = Net('batch')
 model = torch.load("model.pth", map_location=torch.device('cpu'))
+model.eval()
+model.to(torch.device('cpu'))
 
 classes = ('plane', 'car', 'bird', 'cat', 'deer',
            'dog', 'frog', 'horse', 'ship', 'truck')