Update app.py

app.py

@@ -78,7 +78,7 @@ def inference_visualization(input_img, transparency = 0.5, target_layer_number =
 # def gradio_callback(view_gradcam, num_gradcam_images, layer_name, opacity,
 #                     view_misclassified, num_misclassified_images,
 #                         input_img,submit):
-def gradio_callback(view_gradcam_img, input_img, transparency = 0.5, target_layer_number = -1):
+def gradio_callback(input_img, transparency = 0.5, target_layer_number = -1):
     
     confidence = inference_confidences(input_img, transparency = 0.5, target_layer_number = -1)
     visualization = inference_visualization(input_img, transparency = 0.5, target_layer_number = -1)
@@ -108,7 +108,8 @@ demo = gr.Interface(
         # gr.Radio(["Yes", "No"], label="View misclassified images?"),
         # gr.Number(label="Number of misclassified images to view", default=5, min=1, max=10),
         # gr.Image(shape=(32, 32), label="Input Image")
-        gr.Radio(["Yes", "No"], label="View GradCAM images?"),
+        # gr.Radio(["Yes", "No"], label="View GradCAM images?"),
+        gr.Radio(["Yes", "No"], label="Location", info="GradCAM images?"),
         gr.Image(shape=(32, 32), label="Input Image"), 
         gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM"), 
         gr.Slider(-2, -1, value = -2, step=1, label="Which Layer?")