blaise commited on
Commit
a06bf32
1 Parent(s): 591da62

Added requirements and app file

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