Spaces:
Runtime error
Runtime error
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) |