Update app.py

app.py

@@ -107,7 +107,7 @@ demo = gr.Interface(
 # Callback function for the Gradio interface
 def gradio_callback(view_gradcam, num_gradcam_images, layer_name, opacity,
                     view_misclassified, num_misclassified_images,
-                    input_img):
+                        input_img,submit,layer_name):
     confidence = inference_confidences(input_img, transparency = 0.5, target_layer_number = -1)
     visualization = inference_visualization(input_img, transparency = 0.5, target_layer_number = -1)
     return confidence, visualization