Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,6 @@ def get_embedding(image: Image.Image, device="cpu"):
|
|
| 25 |
emb = emb / emb.norm(p=2, dim=-1, keepdim=True)
|
| 26 |
return emb
|
| 27 |
|
| 28 |
-
@spaces.GPU
|
| 29 |
def get_reference_embeddings():
|
| 30 |
if os.path.exists(CACHE_FILE):
|
| 31 |
with open(CACHE_FILE, "rb") as f:
|
|
|
|
| 25 |
emb = emb / emb.norm(p=2, dim=-1, keepdim=True)
|
| 26 |
return emb
|
| 27 |
|
|
|
|
| 28 |
def get_reference_embeddings():
|
| 29 |
if os.path.exists(CACHE_FILE):
|
| 30 |
with open(CACHE_FILE, "rb") as f:
|