Update logic.py

logic.py

@@ -434,7 +434,7 @@ def load_model(train_data, val_data, test_data):
     model = Model(hidden_channels=32)
     with torch.no_grad():
         model.encoder(train_data.x_dict, train_data.edge_index_dict)
-    model.load_state_dict(torch.load('model.pt'))
+    model.load_state_dict(torch.load('model.pt',map_location=torch.device('cpu')))
     model.eval()
     return model