Spaces:

escottyp
/

face-id

Sleeping

escottyp commited on May 16, 2023

Commit

546decd

1 Parent(s): b7740b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,11 +1,11 @@
 import gradio as gr
 import matplotlib.pyplot as plt
-from sentence_transformers import SentenceTransformer
 def predict(im1, im2):
   model = SentenceTransformer('sentence-transformers/clip-ViT-B-16')
   emb = model.encode([im1, im2])
-  sim = cosine_similarity(emb)[0][1]
   if sim > .90:
     return sim, "SAME PERSON, UNLOCK PHONE"
   else:

 import gradio as gr
 import matplotlib.pyplot as plt
+from sentence_transformers import util, SentenceTransformer
 def predict(im1, im2):
   model = SentenceTransformer('sentence-transformers/clip-ViT-B-16')
   emb = model.encode([im1, im2])
+  sim = util.cos_sim(emb)[0][1]
   if sim > .90:
     return sim, "SAME PERSON, UNLOCK PHONE"
   else: