moshel commited on
Commit
efad804
·
1 Parent(s): 176cf5c
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -63,7 +63,11 @@ def predict(inp):
63
  # eigen_smooth=args.eigen_smooth,
64
  # aug_smooth=args.aug_smooth
65
  )
66
- cam_image = show_cam_on_image(img/255, grad[0 :])
 
 
 
 
67
  return confidences,cam_image
68
 
69
  import gradio as gr
 
63
  # eigen_smooth=args.eigen_smooth,
64
  # aug_smooth=args.aug_smooth
65
  )
66
+ img = img/255.0
67
+ grad = grad[0 :]
68
+ print(img.shape, img.dtype)
69
+ print(grad.shape, grad.dtype)
70
+ cam_image = show_cam_on_image(img, grad)
71
  return confidences,cam_image
72
 
73
  import gradio as gr