Spaces:

MagnaSC
/

ImgSearch

Running on Zero

AkinyemiAra commited on Jun 23

Commit

ef821e0

verified ·

1 Parent(s): a4d053b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def get_embedding(image: Image.Image, device="cpu"):
     emb = emb / emb.norm(p=2, dim=-1, keepdim=True)
     return emb
-@space.GPU
 def get_reference_embeddings():
     if os.path.exists(CACHE_FILE):
         with open(CACHE_FILE, "rb") as f:

     emb = emb / emb.norm(p=2, dim=-1, keepdim=True)
     return emb
+@spaces.GPU
 def get_reference_embeddings():
     if os.path.exists(CACHE_FILE):
         with open(CACHE_FILE, "rb") as f: