import gradio as gr
from detector import bat_detector

def find_bats(input_img):
    bats, time = bat_detector(input_img)
    return f"{bats}" + " bats found in " + f"{time:.2f}s"

iface = gr.Interface(
    fn=find_bats, 
    inputs=gr.Image(shape=(640, 512)), 
    outputs='text'
)

iface.launch()