909ahmed commited on
Commit
fcb4bd2
·
verified ·
1 Parent(s): f2be81d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -16,6 +16,7 @@ def process_image_and_text(image, text):
16
 
17
  with torch.no_grad():
18
  image_features = model.encode_image(image)
 
19
  text_features = model.encode_text(text_tokens)
20
 
21
  logits_per_image, logits_per_text = model(image, text_tokens)
 
16
 
17
  with torch.no_grad():
18
  image_features = model.encode_image(image)
19
+ print(image_features.size())
20
  text_features = model.encode_text(text_tokens)
21
 
22
  logits_per_image, logits_per_text = model(image, text_tokens)