Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -102,7 +102,7 @@ def compute_similarity(
|
|
102 |
inputs = processor(image, return_tensors="pt").to(device)
|
103 |
|
104 |
with torch.no_grad():
|
105 |
-
image_emb, _ = img_encoder(inputs.pixel_values)
|
106 |
image_emb = image_emb.to(device, torch.float16)
|
107 |
|
108 |
# Map languages
|
|
|
102 |
inputs = processor(image, return_tensors="pt").to(device)
|
103 |
|
104 |
with torch.no_grad():
|
105 |
+
image_emb, _, _, _ = img_encoder(inputs.pixel_values)
|
106 |
image_emb = image_emb.to(device, torch.float16)
|
107 |
|
108 |
# Map languages
|