Update app.py

app.py

@@ -44,7 +44,9 @@ def get_gradcam_image(input_tensor, label, target_layer):
 
 
 def image_classifier(input_image, top_classes=3, show_cam=True, target_layers=[2, 3], transparency=0.5):
+    orig_image = input_image
     input_image = transform(input_image)
+     
     input_image = input_image.unsqueeze(0)
     output = model(input_image)
     
@@ -62,7 +64,7 @@ def image_classifier(input_image, top_classes=3, show_cam=True, target_layers=[2
     if show_cam:
         for layer in target_layers:
             grayscale_cam = get_gradcam_image(input_image, label, layer)
-            output_image = show_cam_on_image(input_image / 255, grayscale_cam, use_rgb=True, image_weight=transparency)
+            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}"))    
 
     return outputs, confidences