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)
|