peeyushsinghal commited on
Commit
cd6a639
·
1 Parent(s): f2dcc55

updated to show only predicted value for user selected images

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -188,8 +188,10 @@ def process_gradcam_images(num_images,layer,opacity,image_list=None):
188
  rgb_img = rgb_img.numpy()
189
 
190
  visualization = show_cam_on_image(rgb_img, grayscale_cam, use_rgb=True, image_weight=opacity)
191
-
192
- grad_images.append(((visualization),f'Pred: {classes[pred.cpu()]} | Truth :{classes[correct.cpu()]}'))
 
 
193
 
194
  print(str(num_images) + "**" + str(layer) + "**" + str(opacity))
195
  return grad_images, gr.update(visible=True)
 
188
  rgb_img = rgb_img.numpy()
189
 
190
  visualization = show_cam_on_image(rgb_img, grayscale_cam, use_rgb=True, image_weight=opacity)
191
+ if not image_list:
192
+ grad_images.append(((visualization),f'Pred: {classes[pred.cpu()]} | Truth :{classes[correct.cpu()]}'))
193
+ else:
194
+ grad_images.append(((visualization),f'Prediction: {classes[pred.cpu()]}'))
195
 
196
  print(str(num_images) + "**" + str(layer) + "**" + str(opacity))
197
  return grad_images, gr.update(visible=True)