Update README.md
Browse files
README.md
CHANGED
@@ -77,7 +77,7 @@ with torch.no_grad():
|
|
77 |
logits_per_image = outputs.logits_per_image
|
78 |
probs = logits_per_image.softmax(dim=-1)
|
79 |
for t, p in zip(texts, probs[0]):
|
80 |
-
print(f"{p:.1%}
|
81 |
```
|
82 |
Output should look like
|
83 |
```
|
|
|
77 |
logits_per_image = outputs.logits_per_image
|
78 |
probs = logits_per_image.softmax(dim=-1)
|
79 |
for t, p in zip(texts, probs[0]):
|
80 |
+
print(f"'{t}' : {p:.1%}")
|
81 |
```
|
82 |
Output should look like
|
83 |
```
|