Update app.py

app.py

@@ -45,7 +45,7 @@ def inference(input_img, transparancy = 0.5, target_layer_number = -1):
     softmax =  torch.nn.Softmax(dim=0)
     o = softmax(outputs.flattern())
     confidences = {classes[i]:float(o[i] for i in range(10))}
-    - , prediction= torch.max(outputs, 1)
+    prediction= torch.max(outputs, 1)
     target_layers = [model.layer2[target_layer_number]]
     cam = GradCAM(model=model, target_layers=target_layers)
     grayscale_cam = cam(input_tensor= input_img,target=None)