Spaces:
Running
Running
jit fix 5
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ def process_image(image):
|
|
53 |
inputs = processor(images=image, return_tensors="pt")["pixel_values"].to(device)
|
54 |
|
55 |
# Use the traced model for inference
|
56 |
-
outputs = traced_model(
|
57 |
|
58 |
# Normalize the features before search, whatever that means
|
59 |
embeddings = outputs[0].mean(dim=1)
|
|
|
53 |
inputs = processor(images=image, return_tensors="pt")["pixel_values"].to(device)
|
54 |
|
55 |
# Use the traced model for inference
|
56 |
+
outputs = traced_model(inputs)
|
57 |
|
58 |
# Normalize the features before search, whatever that means
|
59 |
embeddings = outputs[0].mean(dim=1)
|