Mitsua commited on
Commit
07e5ffb
1 Parent(s): 9e9ec83

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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%} that image 0 is '{t}'")
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
  ```