Update app.py

app.py

@@ -72,7 +72,7 @@ def inference(input_img, num_gradcam_images=1, target_layer_number=-1, transpare
 
         return top_n_confidences, visualization, visualization_misclassified
     else:
-        return top_n_confidences, visualization
+        return top_n_confidences, visualization, None
 
 title = "CIFAR10 trained on ResNet18 Model with GradCAM"
 description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"