panda1835 commited on
Commit
209677b
·
verified ·
1 Parent(s): ff14199

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -46,9 +46,9 @@ def extract_embedding(image):
46
 
47
  # Get the embedding of the image.
48
  with torch.no_grad():
49
- embedding = dinov2(transformed_img)
50
- # print(embedding.shape)
51
- embedding = embedding[0].cpu().numpy().tolist()
52
  return {
53
  "embedding": embedding
54
  }
 
46
 
47
  # Get the embedding of the image.
48
  with torch.no_grad():
49
+ embedding = dinov2(transformed_img)
50
+ # print(embedding.shape)
51
+ embedding = embedding[0].cpu().numpy().tolist()
52
  return {
53
  "embedding": embedding
54
  }