Update
Browse files
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
|
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
|