escottyp commited on
Commit
4269e73
·
1 Parent(s): 04c7834

create app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import matplotlib.pyplot as plt
3
+ from sentence_transformers import SentenceTransformer
4
+
5
+ def predict(im1, im2):
6
+ model = SentenceTransformer('sentence-transformers/clip-ViT-B-16')
7
+ emb = model.encode([im1, im2])
8
+ sim = cosine_similarity(emb)[0][1]
9
+ if sim > .90:
10
+ return sim, "SAME PERSON, UNLOCK PHONE"
11
+ else:
12
+ return sim, "DIFFERENT PEOPLE, DON'T UNLOCK"
13
+
14
+ interface = gr.Interface(fn=predict,
15
+ inputs= [gr.Image(type="pil", source="webcam"),
16
+ gr.Image(type="pil", source="webcam")],
17
+ outputs= [gr.Number(label="Similarity"),
18
+ gr.Textbox(label="Message")]
19
+ )
20
+
21
+ interface.launch(debug=True)