import gradio as gr
import torch
from transformers import pipeline


model = pipeline('onnx', model='model/best.onnx')

def object_detection(image):
    results = model(image)

    return {
        'image': image,
        'detections': results
    }

inputs = gr.inputs.Image()
outputs = gr.outputs.Image(label='Input Image', type='numpy'), gr.outputs.JSON(label='Object Detections')

iface = gr.Interface(fn=object_detection, inputs=inputs, outputs=outputs)

iface.launch()