jonathan-roos's picture
Fixed timer
1785d30
raw
history blame
No virus
409 Bytes
import gradio as gr
import time
from detector import bat_detector
def find_bats(input_img):
start = time.time()
after_image, total_bats = bat_detector(input_img)
return f"{total_bats}" + " bats found in " + f"{time.time()-start:.2f}s", after_image
iface = gr.Interface(
fn=find_bats,
inputs=gr.Image(shape=(640, 512)),
outputs=['text', gr.Image(shape=(960, 768))]
)
iface.launch()