Add temperature

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import torch
 import torch.nn.functional as F
+from open_clip import create_model, get_tokenizer
 from torchvision import transforms
 
-from open_clip import create_model, get_tokenizer
 from templates import openai_imagenet_template
 
 model_str = "hf-hub:imageomics/bioclip"
@@ -44,7 +44,7 @@ def predict(img, cls_str: str) -> dict[str, float]:
 
     img_features = model.encode_image(img.unsqueeze(0))
     img_features = F.normalize(img_features, dim=-1)
-    logits = (img_features @ txt_features).squeeze()
+    logits = (model.logit_scale.exp() * img_features @ txt_features).squeeze()
     probs = F.softmax(logits, dim=0).tolist()
     return {cls: prob for cls, prob in zip(classes, probs)}
 
@@ -71,4 +71,3 @@ if __name__ == "__main__":
     )
 
     demo.launch()
-