Update app.py
Browse files
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)
|