Update README.md
Browse files
README.md
CHANGED
@@ -90,8 +90,9 @@ Then from any script whose `cwd` is the repos main directory (`cd <tbr>`) run:
|
|
90 |
|
91 |
```python
|
92 |
from PIL import Image
|
93 |
-
from huggingface import Geolocalizer
|
94 |
-
|
|
|
95 |
img = Image.open('.media/examples/img1.jpeg')
|
96 |
x = geoloc.transform(img).unsqueeze(0) # transform the image using our dedicated transformer
|
97 |
gps = geolocalizer(x) # B, 2 (lat, lon - tensor in rad)
|
|
|
90 |
|
91 |
```python
|
92 |
from PIL import Image
|
93 |
+
from models.huggingface import Geolocalizer
|
94 |
+
|
95 |
+
geoloc = Geolocalizer.from_pretrained('osv5m/baseline')
|
96 |
img = Image.open('.media/examples/img1.jpeg')
|
97 |
x = geoloc.transform(img).unsqueeze(0) # transform the image using our dedicated transformer
|
98 |
gps = geolocalizer(x) # B, 2 (lat, lon - tensor in rad)
|