Update app.py

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
 from transformers import CLIPModel, AutoTokenizer, RawImage
-import torch
-import torch.nn.functional as F
 
 # Load the CLIP model and tokenizer
 model = CLIPModel.from_pretrained("Xenova/mobileclip_blt")
@@ -20,7 +18,7 @@ def compute_probability(image):
     text_embeds = model(text_inputs)
     normalized_text_embeds = text_embeds.normalize().tolist()
 
-    probabilities = [F.softmax(torch.tensor([100 * torch.dot(torch.tensor(x), torch.tensor(y)) for y in normalized_text_embeds])).tolist()[0] for x in normalized_image_embeds]
+    probabilities = [softmax([dot(x, y) for y in normalized_text_embeds])[0] for x in normalized_image_embeds]
 
     return {"probability": probabilities[0]}