moshel commited on
Commit
9a56e8d
·
1 Parent(s): c51e43f
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -64,9 +64,8 @@ def predict(inp):
64
  # eigen_smooth=args.eigen_smooth,
65
  # aug_smooth=args.aug_smooth
66
  )
67
- img = img/255.0
68
- grad = grad[0 :]
69
- print(img.shape, img.dtype)
70
  print(grad.shape, grad.dtype)
71
  cam_image = show_cam_on_image(rgb_img, grad)
72
  return confidences,cam_image
 
64
  # eigen_smooth=args.eigen_smooth,
65
  # aug_smooth=args.aug_smooth
66
  )
67
+ grad = grad[0, :]
68
+ print(rgb_img.shape, rgb_img.dtype)
 
69
  print(grad.shape, grad.dtype)
70
  cam_image = show_cam_on_image(rgb_img, grad)
71
  return confidences,cam_image