initial commit

app.py

@@ -0,0 +1,23 @@
+import gradio as gr
+import matplotlib.pyplot as plt
+from sentence_transformers import SentenceTransformer
+from sklearn.metrics.pairwise import cosine_similarity
+
+def predict(im1, im2):
+  model, theta = SentenceTransformer('sentence-transformers/clip-ViT-L-14'), 0.55
+  emb1, emb2 = model.encode(im1), model.encode(im2)
+  sim = cosine_similarity(emb1.reshape(1, -1), emb2.reshape(1, -1))[0][0]
+
+  if sim > theta:
+    return sim, "User authenticated. Phone unlocked"
+  else:
+    return sim, "Unrecognized user. Unable to unlock phone"
+
+interface = gr.Interface(fn=predict, 
+                         inputs= [gr.Image(type="pil", source="webcam"), 
+                                  gr.Image(type="pil", source="webcam")], 
+                         outputs= [gr.Number(label="Similarity"),
+                                   gr.Textbox(label="Message")]
+                         )
+
+interface.launch(debug=True)

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+transformers==4.18.0
+sentence_transformers==2.2.0
+torch==1.11.0