Update demo.py

demo.py

@@ -5,22 +5,22 @@ import io
 import cv2
 import numpy as np
 
-palmprint_count = 0
+palmvein_count = 0
 
-def compare_palmprint(frame1, frame2):
-    global palmprint_count
+def palmvein(frame1, frame2):
+    global palmvein_count
     
-    palmprint_count = palmprint_count + 1
+    palmvein_count = palmvein_count + 1
 
-    url = "http://127.0.0.1:8080/compare_palmprint"
+    url = "http://127.0.0.1:8080/palmvein"
     files = {'file1': open(frame1, 'rb'), 'file2': open(frame2, 'rb')}
 
     r = requests.post(url=url, files=files)
 
     html = None
 
-    compare_result = r.json().get('compare_result')
-    compare_similarity = r.json().get('compare_similarity')
+    compare_result = r.json().get('result')
+    compare_similarity = r.json().get('score')
 
     html = ("<table>"
                 "<tr>"
@@ -42,7 +42,7 @@ def compare_palmprint(frame1, frame2):
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-    # KBY-AI Palm-Print Recognition 
+    # KBY-AI Palm-Vein Recognition 
     We offer SDKs for face recognition, liveness detection(anti-spoofing), ID card recognition and ID document liveness detection. 
     We also specialize in providing outsourcing services with a variety of technical stacks like AI(Computer Vision/Machine Learning), Mobile apps, and web apps.
     
@@ -51,30 +51,30 @@ with gr.Blocks() as demo:
     """
     )
 
-    with gr.TabItem("Palmprint Recognition"):
+    with gr.TabItem("PalmVein Recognition"):
         gr.Markdown(
             """
-        ##### Docker Hub - https://hub.docker.com/r/kbyai/palmprint-recognition
+        ##### Docker Hub - https://hub.docker.com/r/kbyai/palmvein-recognition
         ```bash
-        sudo docker pull kbyai/palmprint-recognition:latest
-        sudo docker run -v ./license.txt:/root/kby-ai-palmprint/license.txt -p 8081:8080 -p 9001:9000 kbyai/palmprint-recognition:latest
+        sudo docker pull kbyai/palmvein-recognition:latest
+        sudo docker run -v ./license.txt:/home/openvino/kby-ai-palmvein/license.txt -p 8081:8080 -p 9001:9000 kbyai/palmvein-recognition:latest
         ```
         """
         )
         with gr.Row():
             with gr.Column():
-                compare_palmprint_input1 = gr.Image(type='filepath')
-                gr.Examples(['palmprint_examples/1.jpg', 'palmprint_examples/3.jpg', 'palmprint_examples/5.jpg', 'palmprint_examples/7.jpg'], 
-                            inputs=compare_palmprint_input1)
-                compare_palmprint_button = gr.Button("Compare Hand")
+                compare_palmvein_input1 = gr.Image(type='filepath')
+                gr.Examples(['palmvein_examples/1.jpg', 'palmvein_examples/3.jpg', 'palmvein_examples/5.jpg', 'palmvein_examples/7.jpg'], 
+                            inputs=compare_palmvein_input1)
+                compare_palmvein_button = gr.Button("Compare Hand")
             with gr.Column():
-                compare_palmprint_input2 = gr.Image(type='filepath')
-                gr.Examples(['palmprint_examples/2.jpg', 'palmprint_examples/4.jpg', 'palmprint_examples/6.jpg', 'palmprint_examples/8.jpg'], 
-                            inputs=compare_palmprint_input2)
+                compare_palmvein_input2 = gr.Image(type='filepath')
+                gr.Examples(['palmvein_examples/2.jpg', 'palmvein_examples/4.jpg', 'palmvein_examples/6.jpg', 'palmvein_examples/8.jpg'], 
+                            inputs=compare_palmvein_input2)
             with gr.Column():
                 compare_result_output = gr.HTML(label='Result')
 
-        compare_palmprint_button.click(compare_palmprint, inputs=[compare_palmprint_input1, compare_palmprint_input2], outputs=[compare_result_output])
+        compare_palmvein_button.click(palmvein, inputs=[compare_palmvein_input1, compare_palmvein_input2], outputs=[compare_result_output])
     gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fweb.kby-ai.com%2F"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fweb.kby-ai.com%2F&label=VISITORS&countColor=%23263759" /></a>')
 
 demo.launch(server_name="0.0.0.0", server_port=7860)