edit mhg checkpoint

models/mhg_model/load.py

@@ -76,9 +76,9 @@ class PretrainedModelWrapper:
 def load(model_name: str = "models/mhg_model/pickles/mhggnn_pretrained_model_0724_2023.pickle") -> Optional[
     PretrainedModelWrapper]:
     repo_id = "ibm/materials.mhg-ged"
-    filename = "mhggnn_pretrained_model_0724_2023.pickle"
+    filename = "pytorch_model.bin" #"mhggnn_pretrained_model_0724_2023.pickle"
     file_path = hf_hub_download(repo_id=repo_id, filename=filename)
     with open(file_path, "rb") as f:
-        model_dict = pickle.load(f)
+        model_dict = torch.load(f)
         return PretrainedModelWrapper(model_dict)
     return None