Update app.py

app.py

@@ -66,6 +66,7 @@ def show_gradcam_images(n, a, b):
     images_with_gradcam = []
     for image_path, label in images:
         image = Image.open(image_path)
+        image = image.resize((32, 32)) 
         image_array = np.asarray(image)
         visualization = inference(image_array, "Yes", a, b)[-1]
         images_with_gradcam.append((visualization, label))
@@ -140,55 +141,56 @@ def change_mygrad_view(choice):
         return grad_or_not.update(visible=False)
 
 
-with gr.Blocks(theme='abidlabs/dracula_revamped') as demo:
+with gr.Blocks(theme='xiaobaiyuan/theme_brief') as demo:
     gr.Markdown("""
     
-    # CustomResNet with GradCAM - Interactive Interface
+    # CustomResNet model with GradCAM 
     
     ### A simple Gradio interface to infer on CustomResNet model and get GradCAM results
     
     """)
-    gr.Markdown("# Analyse the Model")
-    gr.Markdown("## Grad-CAM")
+    #gr.Markdown("# Model")
+    gr.Markdown("## Grad-CAM Images")
     with gr.Row():
       grad_yes_no = gr.Radio(choices = ["Yes", "No"], value="No", label="Do you want to see GradCAM images")
     with gr.Row(visible=False) as grad_block:
         with gr.Column(scale=1):
-            input_grad = gr.Slider(1, 10, value = 3, step=1, label="Number of GradCAM images to view")
+            input_grad = gr.Slider(1, 10, value = 5, step=1, label="Number of GradCAM images to view")
             input_overlay = gr.Radio(choices = ["Yes", "No"], value="No", label="Do you want to configure gradcam")
             with gr.Row():
               clear_btn3 = gr.ClearButton()
               submit_btn3 = gr.Button("Submit")
         with gr.Column(scale=1):
-            input_slider31 = gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM", interactive=True, visible=False)
-            input_slider32 = gr.Slider(1, 3, value = 3, step=1, label="Which Layer?", interactive=True, visible=False)     
+            input_slider1 = gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM", interactive=True, visible=False)
+            input_slider2 = gr.Slider(1, 3, value = 3, step=1, label="Which Layer?", interactive=True, visible=False)     
     with gr.Row(visible=False) as grad1_block:
         gallery3 = gr.Gallery(
                 label="GradCAM images", show_label=True, elem_id="gallery3"
             ).style(columns=[4], rows=[3], object_fit="contain", height="auto")
 
-    submit_btn3.click(fn=show_gradcam_images, inputs=[input_grad, input_slider31, input_slider32], outputs = [grad1_block, gallery3])
-    clear_btn3.click(lambda: [None, None, None, None, None], outputs=[input_grad, input_grad, input_slider31, input_slider32,  gallery3])
-    input_overlay.change(fn=change_textbox, inputs=input_overlay, outputs=[input_slider31, input_slider32])
+    submit_btn3.click(fn=show_gradcam_images, inputs=[input_grad, input_slider1, input_slider2], outputs = [grad1_block, gallery3])
+    clear_btn3.click(lambda: [None, None, None, None, None], outputs=[input_grad, input_grad, input_slider1, input_slider2,  gallery3])
+    input_overlay.change(fn=change_textbox, inputs=input_overlay, outputs=[input_slider1, input_slider2])
     grad_yes_no.change(fn=change_grad_view, inputs=grad_yes_no, outputs=[grad_block])
 
     
     ###############################################
 
     
-    gr.Markdown("## Misclassification")
+    gr.Markdown("## Misclassification Images")
     with gr.Row():
       miscls_yes_no = gr.Radio(choices = ["Yes", "No"], value="No", label="Do you want to see misclassified images")
     with gr.Row(visible=False) as miscls_block:
         with gr.Column(scale=1):
             input_miscn = gr.Slider(1, 10, value = 3, step=1, label="Number of misclassified images to view")
-            with gr.Row():
-              clear_btn2 = gr.ClearButton()
-              submit_btn2 = gr.Button("Submit")
+
         with gr.Column(scale=1):
             input_grad2 = gr.Radio(choices = ["Yes", "No"], value="No", label="Do you want to overlay gradcam")
             input_slider21 = gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM", interactive=True, visible=False)
             input_slider22 = gr.Slider(1, 3, value = 3, step=1, label="Which Layer?", interactive=True, visible=False)    
+            with gr.Row():
+              clear_btn2 = gr.ClearButton()
+              submit_btn2 = gr.Button("Submit")
     with gr.Column(visible=False) as miscls1_block:
         gallery = gr.Gallery(
                 label="Misclassified images", show_label=True, elem_id="gallery"
@@ -204,18 +206,19 @@ with gr.Blocks(theme='abidlabs/dracula_revamped') as demo:
     ###############################################
 
     
-    gr.Markdown("## Try it Out")
+    gr.Markdown("## Input Interface ")
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(shape=(32, 32), label="Input Image")
             input_topk = gr.Slider(1, 10, value = 3, step=1, label="Top N Classes")
             input_slider_grad_or_not =  gr.Radio(choices = ["Yes", "No"], value="No", label="Do you want to overlay GradCAM output")
-            with gr.Column(visible=False) as grad_or_not:
-              input_slider1 = gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM")
-              input_slider2 = gr.Slider(1, 3, value = 3, step=1, label="Which Layer?")
             with gr.Row():
               clear_btn = gr.ClearButton()
               submit_btn = gr.Button("Submit")
+            with gr.Column(visible=False) as grad_or_not:
+              input_slider1 = gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM")
+              input_slider2 = gr.Slider(1, 3, value = 3, step=1, label="Which Layer?")
+
         with gr.Column(scale=1):
             output_classes = gr.Label(num_top_classes=3)
             output_image = gr.Image(shape=(32, 32), label="Output").style(width=128, height=128)