added timer

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from detector import bat_detector
 
 def find_bats(input_img):
-    bats = bat_detector(input_img)
-    return f"{bats}" + " bats found!"
+    bats, time = bat_detector(input_img)
+    return f"{bats}" + " bats found in " + f"{time:.2f}s"
 
 iface = gr.Interface(
     fn=find_bats, 

detector.py

@@ -1,7 +1,7 @@
 import cv2 as cv
 import numpy as np
 from fastai.vision.all import *
-import pathlib
+import time
 
 def chop_img(img, step):
     allImgs = []
@@ -64,6 +64,7 @@ def find_bats(allImgs):
     return cropped_bats
 
 def bat_detector(img):
+    start=time.time()
     allImgs = chop_img(img, 3)
     cropped_bats = find_bats(allImgs)
 
@@ -78,4 +79,4 @@ def bat_detector(img):
 
     filtered_results = filter(lambda x: (x[0] < '0.5' and x[1] > '0.75'), results)
 
-    return len(list(filtered_results))
+    return len(list(filtered_results)), (time.time()-start)