Update logic2.py

logic2.py

@@ -62,11 +62,12 @@ class Model(torch.nn.Module):
 #----------------------------------------------
 def load_hetero_data():
     with open('Hgraph.pkl', 'rb') as file:
+        global data
         data = pickle.load(file)
         return data
 
 def load_model(data):
-    global data
+    
     model = Model(hidden_channels=32)
     with torch.no_grad():
         model.encoder(data.x_dict, data.edge_index_dict)