sanjanatule commited on
Commit
d94b7a6
·
1 Parent(s): 5f16d7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -118,9 +118,7 @@ def inference(input_img, see_misclassified,num_misclassified_imgs,see_gradcam,nu
118
  rgb_img = np.transpose(img, (1, 2, 0))
119
  rgb_img = rgb_img.numpy()
120
  visualization = show_cam_on_image(org_img/255.0, grayscale_cam, use_rgb=True, image_weight=transparency)
121
- plt.imshow(visualization)
122
  else:
123
- plt.imshow(org_img)
124
  visualization = org_img
125
 
126
  # top n classes only
 
118
  rgb_img = np.transpose(img, (1, 2, 0))
119
  rgb_img = rgb_img.numpy()
120
  visualization = show_cam_on_image(org_img/255.0, grayscale_cam, use_rgb=True, image_weight=transparency)
 
121
  else:
 
122
  visualization = org_img
123
 
124
  # top n classes only