Update app.py

app.py

@@ -56,7 +56,7 @@ def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0
     input_img = transform(input_img)
     input_img = input_img.unsqueeze(0)
     outputs = model(input_img)
-    print(outputs)
+    # print(outputs)
     softmax =  torch.nn.Softmax(dim=0)
     o = softmax(outputs.flatten())
     
@@ -72,6 +72,7 @@ def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0
 
     if show_gradcam:
         target_layers = get_layer(layer_name)
+        print(target_layers)
         cam = GradCAM(model=model, target_layers=target_layers)
         grayscale_cam = cam(input_tensor= input_img)
         grayscale_cam = grayscale_cam[0, :]