Update README.md

README.md

@@ -8,6 +8,6 @@ lon_mean = -75.19136178838008
 lon_std = 0.0006865423903444358
 
 <h1>Running Inference</h1>
-model_path = hf_hub_download(repo_id="Latitude-Attitude/vit-gps-coordinates-predictor-with-filter", filename="vit-gps-coordinates-predictor-with-filter-2.pth") model = torch.load(model_path) model.eval()
+model_path = hf_hub_download(repo_id="Latitude-Attitude/vit-gps-coordinates-predictor-with-filter", filename="vit-gps-coordinates-predictor-with-filter-3.pth") model = torch.load(model_path) model.eval()
 
 with torch.no_grad(): for images in dataloader: images = images.to(device) outputs = model(images) preds = outputs.cpu() * torch.tensor([lat_std, lon_std]) + torch.tensor([lat_mean, lon_mean])