Update app.py

app.py

@@ -60,7 +60,7 @@ def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0
     softmax =  torch.nn.Softmax(dim=0)
     o = softmax(outputs.flatten())
     
-    output_numpy = np.squeeze(np.asarray(outputs.numpy()))
+    output_numpy = np.squeeze(np.asarray(outputs.detach().numpy()))
     index_sort = np.argsort(output_numpy)[::-1]
     
     confidences = {}