Update README.md

README.md

@@ -6,8 +6,8 @@ lon_mean = -75.19143495078883
 
 lon_std = 0.0006184167829766685
 
+```
 # TO RUN: 
-
 from huggingface_hub import hf_hub_download
 import torchvision.models as models
 import torch
@@ -16,8 +16,8 @@ import torch.nn as nn
 # Specify the repository and the filename of the model you want to load
 repo_id = "cis-5190-final-fall24/ImageToGPSproject_model"  # Replace with your repo name
 filename = "final_model.pth"
-
 class ResNetGPSModel(nn.Module):
+
     def __init__(self):
         super(ResNetGPSModel, self).__init__()
         self.resnet = models.resnet101()  # Updated for PyTorch >=0.13
@@ -36,4 +36,5 @@ model_path = hf_hub_download(repo_id=repo_id, filename=filename)
 # Load the model using torch
 state_dict = torch.load(model_path)
 model.load_state_dict(state_dict)
-model.eval()  # Set the model to evaluation mode
+model.eval()  # Set the model to evaluation mode
+```