Spaces:
Runtime error
Runtime error
Bastien Dechamps
commited on
Commit
·
989941c
1
Parent(s):
4cf404a
fix path
Browse files
geoguessr_bot/guessr/global_embedder_guessr.py
CHANGED
|
@@ -32,6 +32,7 @@ class GlobalEmbedderGuessr(AbstractGuessr):
|
|
| 32 |
"""Guess a coordinate from an image
|
| 33 |
"""
|
| 34 |
# Embed image
|
|
|
|
| 35 |
image_embedding = self.embedder.embed(image)[None, :]
|
| 36 |
|
| 37 |
# Retrieve nearest neighbors
|
|
|
|
| 32 |
"""Guess a coordinate from an image
|
| 33 |
"""
|
| 34 |
# Embed image
|
| 35 |
+
image = Image.fromarray(image)
|
| 36 |
image_embedding = self.embedder.embed(image)[None, :]
|
| 37 |
|
| 38 |
# Retrieve nearest neighbors
|