methodw commited on
Commit
eaf7419
·
1 Parent(s): 373a0b2
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -56,8 +56,7 @@ def process_image(image):
56
  outputs = traced_model(**inputs)
57
 
58
  # Normalize the features before search, whatever that means
59
- embeddings = outputs.last_hidden_state
60
- embeddings = embeddings.mean(dim=1)
61
  vector = embeddings.detach().cpu().numpy()
62
  vector = np.float32(vector)
63
  faiss.normalize_L2(vector)
 
56
  outputs = traced_model(**inputs)
57
 
58
  # Normalize the features before search, whatever that means
59
+ embeddings = outputs[0].mean(dim=1)
 
60
  vector = embeddings.detach().cpu().numpy()
61
  vector = np.float32(vector)
62
  faiss.normalize_L2(vector)