Update app.py

app.py

@@ -3,18 +3,19 @@ from deepface import DeepFace
 
 def face_similarity(img1, img2):
     result = DeepFace.verify(img1, img2)
-    similarity = result["distance"]  # 낮을수록 더 유사
+    # distance 값을 1에서 빼서 similarity로 사용
+    similarity = 1 - result["distance"]  # 이제 높을수록 더 유사
     verified = result["verified"]
     
-    # 유사도에 따른 메시지 조건 분기
-    if similarity >= 0.5:
+    # 유사도에 따른 메시지 조건 분기, 이제 similarity 값이 높을수록 유사한 것으로 처리
+    if similarity > 0.7:
         message = "두 분은 천생연분!"
-    elif 0.3 <= similarity < 0.5:
+    elif 0.4 < similarity <= 0.7:
         message = "두 분은 내꺼인듯 내꺼아닌 내꺼같은...?"
     else:
         message = "오히려 좋아! 정반대의 사람이 끌릴지도 몰라요"
     
-    return f"유사도: {similarity}, 매칭 결과: {verified}\n{message}"
+    return f"유사도: {similarity:.2f}, 매칭 결과: {verified}\n{message}"
 
 iface = gr.Interface(fn=face_similarity, 
                      inputs=[gr.inputs.Image(shape=(224, 224)), gr.inputs.Image(shape=(224, 224))], 
@@ -24,6 +25,32 @@ iface = gr.Interface(fn=face_similarity,
 
 iface.launch()
 
+# import gradio as gr
+# from deepface import DeepFace
+
+# def face_similarity(img1, img2):
+#     result = DeepFace.verify(img1, img2)
+#     similarity = result["distance"]  # 낮을수록 더 유사
+#     verified = result["verified"]
+    
+#     # 유사도에 따른 메시지 조건 분기
+#     if similarity >= 0.5:
+#         message = "두 분은 천생연분!"
+#     elif 0.3 <= similarity < 0.5:
+#         message = "두 분은 내꺼인듯 내꺼아닌 내꺼같은...?"
+#     else:
+#         message = "오히려 좋아! 정반대의 사람이 끌릴지도 몰라요"
+    
+#     return f"유사도: {similarity}, 매칭 결과: {verified}\n{message}"
+
+# iface = gr.Interface(fn=face_similarity, 
+#                      inputs=[gr.inputs.Image(shape=(224, 224)), gr.inputs.Image(shape=(224, 224))], 
+#                      outputs="text",
+#                      title="얼굴 유사도 측정",
+#                      description="두 얼굴 이미지를 업로드해서 유사도를 측정합니다.")
+
+# iface.launch()
+
 
 
 # import gradio as gr