Spaces:
Build error
Build error
| from saliency import * | |
| from visualizeer import * | |
| def processor(model, image_path, device): | |
| model.load_state_dict(torch.load('./model.pth', map_location=device, weights_only=True)) | |
| model.eval() | |
| s_image = generate_saliency(model, image_path,device) | |
| g_image= overlay_heatmap(image_path, model, device) | |
| b_image = overlay_heatmap1(image_path, model, device) | |
| return s_image, b_image, g_image |