Update app.py

app.py

@@ -66,7 +66,7 @@ def image_classifier(input_image, top_classes=3, show_cam=True, target_layers=[2
         for layer in target_layers:
             grayscale_cam = get_gradcam_image(input_image, label, layer)
             output_image = show_cam_on_image(orig_image / 255, grayscale_cam, use_rgb=True, image_weight=transparency)
-            outputs.append((output_image, f"Layer {layer - 4}"))    
+            outputs.append((output_image, f"Layer {layer - 5}"))    
 
     return outputs, confidences
 
@@ -93,8 +93,8 @@ def show_incorrect(num_examples=10):
     for i in range(num_examples):
         j = np.random.randint(0,wrong_img_no)
         image = np.asarray(Image.open(f'misclassified-images/{j}.jpg'))
-        actual = classes(wrong_img['actual'][j])
-        predicted = classes(wrong_img['predicted'][j])
+        actual = classes(wrong_imgloc[0].at["actual"])
+        predicted = classes(wrong_img.loc[0].at["predicted"])
         
         result.append((image, f"{actual} / {predicted}"))