jonathan-roos's picture
added output image
e94fae2
raw
history blame
360 Bytes
import gradio as gr
from detector import bat_detector
def find_bats(input_img):
after_image, total_bats = bat_detector(input_img)
return f"{total_bats}" + " bats found in " + f"{time:.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()