osv5m commited on
Commit
717007c
·
1 Parent(s): 32bed39

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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
- Geolocalizer.from_pretrained('osv5m/baseline')
 
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)