Duplicate from brendenc/Face-ID

Co-authored-by: Brenden Connors <brendenc@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 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
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet 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
+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
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Face ID
+emoji: ⚡
+colorFrom: green
+colorTo: red
+sdk: gradio
+sdk_version: 3.4.1
+app_file: app.py
+pinned: false
+duplicated_from: brendenc/Face-ID
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,29 @@
+from sklearn.metrics.pairwise import cosine_similarity
+from sentence_transformers import SentenceTransformer
+import datasets
+import gradio as gr
+
+model = SentenceTransformer('clip-ViT-B-16')
+dataset = datasets.load_dataset('brendenc/celeb-identities')
+
+def predict(im1, im2):
+  
+  embeddings = model.encode([im1, im2])
+  sim = cosine_similarity(embeddings)
+  sim = sim[0, 1]
+  if sim > 0.75:
+    return sim, "SAME PERSON, UNLOCK PHONE"
+  else:
+    return sim, "DIFFERENT PEOPLE, DON'T UNLOCK"
+
+
+interface = gr.Interface(fn=predict, 
+                         inputs= [gr.Image(value = dataset['train']['image'][0], type="pil", source="webcam"), 
+                                  gr.Image(value = dataset['train']['image'][1], type="pil", source="webcam")], 
+                         outputs= [gr.Number(label="Similarity"),
+                                   gr.Textbox(label="Message")],
+                         title = 'Face ID',
+                         description = 'This app uses emage embeddings and cosine similarity to function as a Face ID application. Cosine similarity is used, so it ranges from -1 to 1.'
+                         )
+
+interface.launch(debug=True)

requirements.txt

@@ -0,0 +1,3 @@
+scikit-learn
+datasets
+sentence_transformers