taesiri commited on
Commit
ba78c79
·
1 Parent(s): 3ccdd83
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def calculate_score(image, text, model_name):
49
  cosine_similarities = torch.mm(text_embeds, image_embeds.t()).squeeze(1)
50
 
51
  # Convert to percentages
52
- percentages = ((cosine_similarities + 1) / 2 * 100).cpu().numpy()
53
 
54
  results_dict = {label: float(score) for label, score in zip(labels, percentages)}
55
  return results_dict
 
49
  cosine_similarities = torch.mm(text_embeds, image_embeds.t()).squeeze(1)
50
 
51
  # Convert to percentages
52
+ percentages = ((cosine_similarities + 1) / 2).cpu().numpy()
53
 
54
  results_dict = {label: float(score) for label, score in zip(labels, percentages)}
55
  return results_dict