rounding

app.py

@@ -8,6 +8,7 @@ model, preprocess = clip.load("ViT-B/32", device=device)
 
 def hotornot(image, gender):
     image = Image.fromarray(image.astype("uint8"), "RGB")
+
     image = preprocess(image).unsqueeze(0).to(device)
     positive_terms = [f'a hot {gender}', f'a beautiful {gender}', f'an attractive {gender}']
     negative_terms = [f'a gross {gender}', f'an ugly {gender}', f'a hideous {gender}']
@@ -27,9 +28,9 @@ def hotornot(image, gender):
     positive_probs = [prob[0] for prob in probs]
     negative_probs = [prob[1] for prob in probs]
 
-    hotness_score = (probs[0][0] - probs[0][1] + 1) * 50
-    beauty_score = (probs[1][0] - probs[1][1] + 1) * 50
-    attractiveness_score = (probs[2][0] - probs[2][1] + 1) * 50
+    hotness_score = round((probs[0][0] - probs[0][1] + 1) * 50, 2)
+    beauty_score = round((probs[1][0] - probs[1][1] + 1) * 50, 2)
+    attractiveness_score = round((probs[2][0] - probs[2][1] + 1) * 50, 2)
 
     hot_score = sum(positive_probs)/len(positive_probs)
     ugly_score = sum(negative_probs)/len(negative_probs)