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 |
+
```
|