Hugging Face's logo

Spaces:
AkashDataScience
/
shirt-detection
Sleeping

App Files Community
AkashDataScience commited on
Commit
848cadc
·
1 Parent(s): 2d2f4f1

Minor fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -112,7 +112,7 @@ def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True,
112
  else:
113
  misclassified_images = None
114
 
115
- cam = EigenCAM(model, target_layers, use_cuda=False)
116
  grayscale_cam = cam(rgb_img)[0, :, :]
117
  cam_image = show_cam_on_image(im_resized, grayscale_cam, use_rgb=True)
118
 
 
112
  else:
113
  misclassified_images = None
114
 
115
+ cam = EigenCAM(model, target_layers)
116
  grayscale_cam = cam(rgb_img)[0, :, :]
117
  cam_image = show_cam_on_image(im_resized, grayscale_cam, use_rgb=True)
118