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)
|