more data!

app.py

@@ -27,15 +27,15 @@ def hotornot(image, gender):
     positive_probs = [prob[0] for prob in probs]
     negative_probs = [prob[1] for prob in probs]
 
-    print("+:", positive_probs)
-    print("-:", negative_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
 
     hot_score = sum(positive_probs)/len(positive_probs)
     ugly_score = sum(negative_probs)/len(negative_probs)
-    print(hot_score, ugly_score)
     composite = ((hot_score - ugly_score)+1) * 50
     composite = round(composite, 2)
-    return composite
+    return composite, hotness_score, beauty_score, attractiveness_score
 
 iface = gr.Interface(
     fn=hotornot,
@@ -48,7 +48,12 @@ iface = gr.Interface(
             default='person',
         )
     ],
-    outputs=gr.Textbox(label="Hotness Score"),
+    outputs=[
+        gr.Textbox(label="Total Hot or Not™ Score"),
+        gr.Textbox(label="Hotness Score"),
+        gr.Textbox(label="Beauty Score"),
+        gr.Textbox(label="Attractiveness Score"),
+    ],
     title="Hot or Not",
     description="A simple hot or not app using OpenAI's CLIP model.",
 )