Spaces:
Runtime error
Runtime error
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] | |
def det_image(img, model_path, conf_thres, iou_thres): | |
model.conf = conf_thres | |
model.iou = iou_thres | |
return model(img).render()[0] | |
gr.Interface(inputs=["image", gr.inputs.Slider(minimum=0, maximum=1, default=base_conf), gr.inputs.Slider(minimum=0, maximum=1, default=base_iou)], | |
outputs=["image"], | |
#gr.Interface(inputs=["image", gr.inputs.Dropdown(model_paths, label="选择模型"), 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(share=False) |