import gradio as gr
from inference import *
def inference(img):
    img = cv2.imread(img.name)
    boxes, scores, cls_ids = prediction(img)
    return vis(img, boxes, scores, cls_ids)
examples = [['input.jpg']]
gr.Interface(inference, inputs=gr.inputs.Image(type="file"), outputs=gr.outputs.Image(type="pil"),
    examples=examples).launch()