Update demo.py

demo.py

@@ -4,9 +4,9 @@ import datadog_api_client
 from PIL import Image
 
 
-def idcard_recognition(frame):
-    url = "http://127.0.0.1:8080/idcard_recognition"
-    files = {'file': open(frame, 'rb')}
+def idcard_recognition(frame1, frame2):
+    url = "http://127.0.0.1:8080/idcard_recognition_multi_page"
+    files = {'file1': open(frame1, 'rb'), 'file2': open(frame2, 'rb')}
 
     r = requests.post(url=url, files=files)
 
@@ -115,7 +115,8 @@ with gr.Blocks() as demo:
         )
         with gr.Row():
             with gr.Column(scale=3):
-                id_image_input = gr.Image(type='filepath')
+                id_image_input1 = gr.Image(type='filepath')
+                id_image_input2 = gr.Image(type='filepath')
                 gr.Examples(['idcard_examples/1.jpg', 'idcard_examples/2.jpg', 'idcard_examples/3.jpg'], 
                             inputs=id_image_input)
                 id_recognition_button = gr.Button("ID Card Recognition")
@@ -125,7 +126,7 @@ with gr.Blocks() as demo:
             with gr.Column(scale=2):
                 image_result_output = gr.HTML()
 
-        id_recognition_button.click(idcard_recognition, inputs=id_image_input, outputs=[id_result_output, image_result_output])
+        id_recognition_button.click(idcard_recognition, inputs=[id_image_input1, id_image_input2], outputs=[id_result_output, image_result_output])
     gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fkby-ai%2FIDCardRecognition"><img src="https://api.visitorbadge.io/api/combined?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fkby-ai%2FIDCardRecognition&countColor=%23263759" /></a>')
 
 demo.launch(server_name="0.0.0.0", server_port=7860)