Spaces:
Configuration error
Configuration error
Update README.md
Browse files
README.md
CHANGED
|
@@ -7,4 +7,24 @@ sdk: static
|
|
| 7 |
pinned: false
|
| 8 |
---
|
| 9 |
|
| 10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
pinned: false
|
| 8 |
---
|
| 9 |
|
| 10 |
+
Dataset stats: \
|
| 11 |
+
lat_mean = 39.951564548022596 \
|
| 12 |
+
lat_std = 0.0006361722351128644 \
|
| 13 |
+
lon_mean = -75.19150880602636 \
|
| 14 |
+
lon_std = 0.000611411894337979
|
| 15 |
+
|
| 16 |
+
The model can be loaded using:
|
| 17 |
+
```
|
| 18 |
+
from huggingface_hub import hf_hub_download
|
| 19 |
+
import torch
|
| 20 |
+
|
| 21 |
+
# Specify the repository and the filename of the model you want to load
|
| 22 |
+
repo_id = "FinalProj5190/ImageToGPSproject-vit-base" # Replace with your repo name
|
| 23 |
+
filename = "resnet_gps_regressor_complete.pth"
|
| 24 |
+
|
| 25 |
+
model_path = hf_hub_download(repo_id=repo_id, filename=filename)
|
| 26 |
+
|
| 27 |
+
# Load the model using torch
|
| 28 |
+
model_test = torch.load(model_path)
|
| 29 |
+
model_test.eval() # Set the model to evaluation mode
|
| 30 |
+
```
|