Upload 4 files

.gitattributes

@@ -2,13 +2,11 @@
 *.arrow filter=lfs diff=lfs merge=lfs -text
 *.bin filter=lfs diff=lfs merge=lfs -text
 *.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
 *.ftz filter=lfs diff=lfs merge=lfs -text
 *.gz filter=lfs diff=lfs merge=lfs -text
 *.h5 filter=lfs diff=lfs merge=lfs -text
 *.joblib filter=lfs diff=lfs merge=lfs -text
 *.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
 *.model filter=lfs diff=lfs merge=lfs -text
 *.msgpack filter=lfs diff=lfs merge=lfs -text
 *.npy filter=lfs diff=lfs merge=lfs -text
@@ -16,13 +14,12 @@
 *.onnx filter=lfs diff=lfs merge=lfs -text
 *.ot filter=lfs diff=lfs merge=lfs -text
 *.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
 *.pickle filter=lfs diff=lfs merge=lfs -text
 *.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
 *.pt filter=lfs diff=lfs merge=lfs -text
 *.pth filter=lfs diff=lfs merge=lfs -text
 *.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.tar.* filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,13 +1,12 @@
 ---
-title: BeautyRate
-emoji: 🐨
-colorFrom: pink
-colorTo: red
+title: Hot Or Not
+emoji: 🏢
+colorFrom: yellow
+colorTo: gray
 sdk: gradio
-sdk_version: 3.29.0
+sdk_version: 3.4.1
 app_file: app.py
 pinned: false
-license: openrail
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,61 @@
+import torch
+import clip
+from PIL import Image
+import gradio as gr
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+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}']
+
+    pairs = list(zip(positive_terms, negative_terms))
+
+    def evaluate(terms):
+        text = clip.tokenize(terms).to(device)
+
+        with torch.no_grad():
+            logits_per_image, logits_per_text = model(image, text)
+            probs = logits_per_image.softmax(dim=-1).cpu().numpy()
+            return probs[0]
+
+    probs = [evaluate(pair) for pair in pairs]
+    
+    positive_probs = [prob[0] for prob in probs]
+    negative_probs = [prob[1] for prob in probs]
+
+    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)
+    composite = ((hot_score - ugly_score)+1) * 50
+    composite = round(composite, 2)
+    return composite, hotness_score, beauty_score, attractiveness_score
+
+iface = gr.Interface(
+    fn=hotornot,
+    inputs=[
+        gr.inputs.Image(label="Image"),
+        gr.inputs.Dropdown(
+            [
+                'person', 'man', 'woman'
+            ],
+            default='person',
+        )
+    ],
+    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. How it works: the input image is passed to OpenAI's CLIP image captioning model and evaluated for how much it conforms to the model's idea of hotness, beauty, and attractiveness. These values are then combined to produce a composite score on a scale of 0 to 100.",
+)
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+torch
+git+https://github.com/openai/CLIP.git
+Pillow
+gradio