Update README.md

README.md

@@ -77,7 +77,8 @@ with torch.no_grad():
   image_feature = image_feature / image_feature.norm(p=2, dim=-1, keepdim=True)
   text_feature = text_feature / text_feature.norm(p=2, dim=-1, keepdim=True)
 
-logits_per_image = image_feature @ text_feature.T 
+logits_per_image = image_feature @ text_feature.T
+logits_per_image = model.logit_scale.exp() * logits_per_image
 probs = logits_per_image.softmax(dim=1) 
 print(probs)