Update main README with correct .pt file extensions

README.md

@@ -95,7 +95,7 @@ model = DJMGNN(
 
 # Load base model checkpoint
 checkpoint = torch.hub.load_state_dict_from_url(
-    "https://huggingface.co/saketh11/MoML-CA/resolve/main/base_model/pytorch_model.bin"
+    "https://huggingface.co/saketh11/MoML-CA/resolve/main/base_model/pytorch_model.pt"
 )
 model.load_state_dict(checkpoint["model_state_dict"])
 model.eval()
@@ -106,7 +106,7 @@ model.eval()
 ```python
 # Same architecture setup as above, then:
 checkpoint = torch.hub.load_state_dict_from_url(
-    "https://huggingface.co/saketh11/MoML-CA/resolve/main/finetuned_model/pytorch_model.bin"
+    "https://huggingface.co/saketh11/MoML-CA/resolve/main/finetuned_model/pytorch_model.pt"
 )
 model.load_state_dict(checkpoint["model_state_dict"])
 model.eval()