Fix GNN model loading to specify CPU mapping for compatibility

src/demo.py

@@ -59,7 +59,7 @@ if "setup_complete" not in st.session_state:
                 dropout_rate=0.5,
             )
             no_edge_gnn_model.load_state_dict(
-                torch.load(config.GNN_MODEL_PATH.replace("undirected_gnn", "no_edge_gnn"))
+                torch.load(config.GNN_MODEL_PATH.replace("undirected_gnn", "no_edge_gnn"), map_location=torch.device("cpu"))
             )
             st.session_state.no_edge_gnn_model = no_edge_gnn_model
             st.session_state.no_edge_graph_data = no_edge_graph_data