app.py

import torch
import gradio as gr

model = torch.hub.load('./', 'custom', path="best.pt", source='local',force_reload=True)


title = "                                               焊缝缺陷检测                                                      "

desc = ""

base_conf, base_iou = 0.30, 0.45

def det_image(img, conf_thres, iou_thres):
    model.conf = conf_thres
    model.iou = iou_thres
    return model(img).render()[0]

gr.Interface(inputs=["image", gr.Slider(minimum=0,maximum=1, value=base_conf), gr.Slider(minimum=0,maximum=1, value=base_iou)],
             outputs=["image"],
             fn=det_image,
             title=title,
             description=desc).launch()