Spaces:
Runtime error
Runtime error
| import torch | |
| import gradio as gr | |
| import matplotlib.pyplot as plt | |
| from sentence_transformers import SentenceTransformer | |
| from sklearn.metrics.pairwise import cosine_similarity | |
| model = SentenceTransformer('clip-ViT-B-32') | |
| def predict(im1, im2): | |
| embeding = model.encode([im1, im2]) | |
| sim = cosine_similarity(embeding) | |
| sim = sim[0][1] | |
| if sim > 0.78: # THRESHOLD HERE | |
| return sim, "SAME PERSON, UNLOCK PHONE" | |
| else: | |
| return sim, "DIFFERENT PEOPLE, DON'T UNLOCK" | |
| import gradio as gr | |
| title = 'Face ID' | |
| description = 'This model detects the similarity between two images and passes a command!' | |
| article = """ | |
| Upload and Image from your Device or Make use of your webcam | |
| """ | |
| img_upload = gr.Interface( | |
| fn=predict, | |
| inputs= [gr.Image(type="pil", source="upload"), | |
| gr.Image(type="pil", source="upload")], | |
| outputs= [gr.Number(label="Similarity"), | |
| gr.Textbox(label="Message")], | |
| title=title, | |
| description=description, | |
| article=article | |
| ) | |
| webcam_upload = 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")], | |
| title=title, | |
| description=description, | |
| article=article, | |
| ) | |
| face_id = gr.TabbedInterface( | |
| [img_upload, webcam_upload], | |
| ["Upload-Image", "Use Webcam"]) | |
| face_id.launch(debug=True) | |