Update app.py

app.py

@@ -23,17 +23,17 @@ from codes import *
 from torchvision.models import resnet18
 
 
-device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+#device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
 file_savingfolder = './modelbest/'
 ext = '_bestVal'
 core_model_tst = resnet18(pretrained=True)
 core_model_tst.fc = Identity()
 core_model_tst.load_state_dict(torch.load(file_savingfolder+'core_model'+ext+'.pth'))
-core_model_tst.to(device)
+#core_model_tst.to(device)
 IR_Model_tst = Build_IRmodel_Resnet(core_model_tst, registration_method)
 IR_Model_tst.load_state_dict(torch.load(file_savingfolder+'IR_Model'+ext+'.pth'))
-IR_Model_tst.to(device)
+#IR_Model_tst.to(device)
 IR_Model_tst.eval()