update on app.py

app.py

@@ -7,7 +7,7 @@ from resnet18 import ResNet18
 
 model = ResNet18(1, 5)
 
-checkpoint = torch.load('acc=0.94.ckpt')
+checkpoint = torch.load('acc=0.94.ckpt', map_location=torch.device('cpu'))
 
 # The state dict will contains net.layer_name
 # Our model doesn't contains `net.` so we have to rename it