Spaces:
Build error
Build error
jagilley
commited on
Commit
•
f263f52
1
Parent(s):
b041e49
rounding
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ model, preprocess = clip.load("ViT-B/32", device=device)
|
|
8 |
|
9 |
def hotornot(image, gender):
|
10 |
image = Image.fromarray(image.astype("uint8"), "RGB")
|
|
|
11 |
image = preprocess(image).unsqueeze(0).to(device)
|
12 |
positive_terms = [f'a hot {gender}', f'a beautiful {gender}', f'an attractive {gender}']
|
13 |
negative_terms = [f'a gross {gender}', f'an ugly {gender}', f'a hideous {gender}']
|
@@ -27,9 +28,9 @@ def hotornot(image, gender):
|
|
27 |
positive_probs = [prob[0] for prob in probs]
|
28 |
negative_probs = [prob[1] for prob in probs]
|
29 |
|
30 |
-
hotness_score = (probs[0][0] - probs[0][1] + 1) * 50
|
31 |
-
beauty_score = (probs[1][0] - probs[1][1] + 1) * 50
|
32 |
-
attractiveness_score = (probs[2][0] - probs[2][1] + 1) * 50
|
33 |
|
34 |
hot_score = sum(positive_probs)/len(positive_probs)
|
35 |
ugly_score = sum(negative_probs)/len(negative_probs)
|
|
|
8 |
|
9 |
def hotornot(image, gender):
|
10 |
image = Image.fromarray(image.astype("uint8"), "RGB")
|
11 |
+
|
12 |
image = preprocess(image).unsqueeze(0).to(device)
|
13 |
positive_terms = [f'a hot {gender}', f'a beautiful {gender}', f'an attractive {gender}']
|
14 |
negative_terms = [f'a gross {gender}', f'an ugly {gender}', f'a hideous {gender}']
|
|
|
28 |
positive_probs = [prob[0] for prob in probs]
|
29 |
negative_probs = [prob[1] for prob in probs]
|
30 |
|
31 |
+
hotness_score = round((probs[0][0] - probs[0][1] + 1) * 50, 2)
|
32 |
+
beauty_score = round((probs[1][0] - probs[1][1] + 1) * 50, 2)
|
33 |
+
attractiveness_score = round((probs[2][0] - probs[2][1] + 1) * 50, 2)
|
34 |
|
35 |
hot_score = sum(positive_probs)/len(positive_probs)
|
36 |
ugly_score = sum(negative_probs)/len(negative_probs)
|