HugoHE's picture
Update app.py
c0dda32
raw
history blame contribute delete
342 Bytes
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()