Spaces:
Running
Running
fix JIT 3
Browse files
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.
|
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)
|