info

app.py

@@ -6,6 +6,8 @@ import math
 import gradio as gr
 
 
+processing_time = ''
+
 def find_signature_bounding_boxes(image):
     # Start measuring time
     start_time = time.time()
@@ -67,6 +69,7 @@ def find_signature_bounding_boxes(image):
     # End measuring time
     end_time = time.time()
     print(f"Function took {end_time - start_time:.2f} seconds to process the image.")
+    processing_time = f"Detection took {end_time - start_time:.2f} seconds. "
 
     return possible_signatures
 
@@ -125,8 +128,8 @@ def run_detection(input_image):
         cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)
     image_with_box = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
     
-
-    return image_with_box
+    info = processing_time + 'Nr of signatures found: ' + str(len(signatures))
+    return image_with_box, info
 
 if __name__ == "__main__":
     
@@ -155,11 +158,11 @@ if __name__ == "__main__":
                 gallery = gr.outputs.Image(type="pil").style(full_width=True, full_height=True)
 
         run_button.click(fn=run_detection, inputs=[
-                        input_image], outputs=[gallery])
+                        input_image], outputs=[gallery, info])
         gr.Examples(
           [["Sample1.jpg"],["Sample2.jpg"]],
           inputs = [input_image],
-          outputs = [gallery],
+          outputs = [gallery, info],
           fn=run_detection,
           cache_examples=True,
           label='Try these examples:'