Update app.py

app.py

@@ -40,20 +40,30 @@ def extract_face(im):
     return file_name
     
 def predict(im1, im2,thresh,model_name):
-  im1_face = Image.open(im1)
-  im2_face = Image.open(im2)
 
+  if not isinstance(im1,str):
+    im1_face = im1
+    im2_face = im2
+
+  else:
+
+    im1_face = Image.open(im1)
+    im2_face = Image.open(im2)
+    
   model = load_model(model_name)
 
   sim=cosine_similarity(model.encode([im1_face,im2_face]))[0][1]
 
   if sim > thresh:
-    return sim, "SAME PERSON, UNLOCK PHONE"
+    return round(sim,2), "SAME PERSON, UNLOCK PHONE"
   else:
-    return sim, "DIFFERENT PEOPLE, DON'T UNLOCK"
+    return round(sim,2), "DIFFERENT PEOPLE, DON'T UNLOCK"
 
 def load_model(model_name):
+
   model = SentenceTransformer(model_name)
+  
+  return model
 
 title = """<h1 id="title">FaceID for Facial Recognition with Face Detector</h1>"""
 
@@ -114,13 +124,5 @@ with demo:
             
     nd_but.click(predict,inputs=[nd_image_input_1,nd_image_input_2,thresh,model_options],outputs=[sim,msg],queue=True)
     fd_but.click(predict,inputs=[face_1,face_2,thresh,model_options],outputs=[sim_1,msg_1],queue=True)
-# 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)
 
 demo.launch(debug=True,enable_queue=True)