FrancoMango
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -26,7 +26,7 @@ class ViTGPSModel(nn.Module):
|
|
26 |
<h1>Running Inference</h1>
|
27 |
|
28 |
```
|
29 |
-
model_path = hf_hub_download(repo_id="Latitude-Attitude/vit-gps-coordinates-predictor-with-filter", filename="vit-gps-coordinates-predictor-with-filter-
|
30 |
model = torch.load(model_path)
|
31 |
model.eval()
|
32 |
|
|
|
26 |
<h1>Running Inference</h1>
|
27 |
|
28 |
```
|
29 |
+
model_path = hf_hub_download(repo_id="Latitude-Attitude/vit-gps-coordinates-predictor-with-filter", filename="vit-gps-coordinates-predictor-with-filter-6.pth")
|
30 |
model = torch.load(model_path)
|
31 |
model.eval()
|
32 |
|