Sibgat-Ul commited on
Commit
565bf9a
·
verified ·
1 Parent(s): 7b3b5c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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