AkashDataScience commited on
Commit
da9824e
·
1 Parent(s): 231fd74
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -112,7 +112,7 @@ def inference(input_img, conf_thres, iou_thres, is_false_detection_images=True,
112
 
113
  cam = EigenCAM(model, target_layers)
114
  grayscale_cam = cam(tensor)[0, :, :]
115
- cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True)
116
 
117
  return img0, cam_image, misclassified_images
118
 
@@ -136,7 +136,7 @@ demo = gr.Interface(inference,
136
  gr.Checkbox(label="Show False Detection"),
137
  gr.Slider(5, 35, value=10, step=5, label="Number of False Detection")],
138
  outputs= [gr.Image(width=640, height=640, label="Output"),
139
- gr.Plot("EigenCAM"),
140
  gr.Plot(label="False Detection")],
141
  title=title,
142
  description=description,
 
112
 
113
  cam = EigenCAM(model, target_layers)
114
  grayscale_cam = cam(tensor)[0, :, :]
115
+ cam_image = show_cam_on_image(img_GC, grayscale_cam, use_rgb=True)
116
 
117
  return img0, cam_image, misclassified_images
118
 
 
136
  gr.Checkbox(label="Show False Detection"),
137
  gr.Slider(5, 35, value=10, step=5, label="Number of False Detection")],
138
  outputs= [gr.Image(width=640, height=640, label="Output"),
139
+ gr.Plot(label="EigenCAM"),
140
  gr.Plot(label="False Detection")],
141
  title=title,
142
  description=description,