Update app.py

app.py

@@ -103,7 +103,7 @@ with gr.Blocks() as app:
     with gr.Row(visible=True) as top_pred_cls_col:
         with gr.Column():
             example_images = gr.Gallery(allow_preview=False, label='Select image ', info='',
-                                        value=[img[0] for img in sample_images], columns=7, rows=2)
+                                        value=[img[0] for img in sample_images], columns=6, rows=2)
 
         with gr.Column():
             with gr.Row():
@@ -111,7 +111,7 @@ with gr.Blocks() as app:
 
             with gr.Row():
                 if_show_grad_cam = gr.Checkbox(value=True, label='Show Class Activation Map (What the model sees)?')
-                
+
             with gr.Row():
                 top_class_btn = gr.Button("Submit", variant='primary')
                 tc_clear_btn = gr.ClearButton()
@@ -226,10 +226,9 @@ with gr.Blocks() as app:
         if not show_cam:
             return [plotted_img, None]
 
-        grayscale_cam = cam(transformed_image)[0, :, :]
-        img = cv2.resize(transformed_image, (cfg.IMAGE_SIZE, cfg.IMAGE_SIZE))
-        #img = np.float32(img) / 255
-        cam_image = show_cam_on_image(img, grayscale_cam, use_rgb=True, image_weight=transparency)
+        grayscale_cam = cam(transformed_image)
+        img = np.array(transformed_image[0], np.float16).transpose(1, 2, 0)
+        cam_image = show_cam_on_image(img, grayscale_cam.transpose(1, 2, 0), use_rgb=True, image_weight=transparency)
         return [plotted_img, cam_image]