Update app.py

app.py

@@ -90,7 +90,7 @@ if __name__ =='__main__':
     
     print('Number fo parameters: {}'.format(model.num_parameters()))
     model.to(device)
-    model_dict = torch.load('Model_example.pth.tar')['state_dict']
+    model_dict = torch.load('Model_example.pth.tar', map_location=device)['state_dict']
     # save model
     model.load_state_dict(model_dict, strict=True)
     model.eval()