Update app.py

app.py

@@ -100,26 +100,26 @@ with gr.Blocks() as app:
             </p>
             """
         )
-    with gr.Row(visible=True) as top_pred_cls_col:
+    with gr.Row(visible=True) as pred_cls_col:
         with gr.Column():
             example_images = gr.Gallery(allow_preview=False, label='Select image ', info='',
                                         value=[img[0] for img in sample_images], columns=6, rows=2)
 
         with gr.Column():
             with gr.Row():
-                top_pred_image = gr.Image(label='Upload Image or Select from the gallery')
+                pred_image = gr.Image(label='Upload Image or Select from the gallery')
 
             with gr.Row():
                 if_show_grad_cam = gr.Checkbox(value=True, label='Show Class Activation Map (What the model sees)?')
 
             with gr.Row():
-                top_class_btn = gr.Button("Submit", variant='primary')
-                tc_clear_btn = gr.ClearButton()
+                submit_btn = gr.Button("Submit", variant='primary')
+                clear_btn = gr.ClearButton()
 
-    with gr.Row(visible=True) as top_class_output:
-        with gr.Column(visible=True)  as top_class_output:
-            top_class_output_img = gr.Image(interactive=False, label='Prediction Output')
-        with gr.Column(visible=True)  as top_class_output:
+    with gr.Row(visible=True) as output_bk:
+        with gr.Column(visible=True)  as output_bk:
+            output_img = gr.Image(interactive=False, label='Prediction Output')
+        with gr.Column(visible=True)  as output_bk:
             grad_cam_out = gr.Image(interactive=False, visible=True, label='CAM Outcome')
 
 
@@ -138,21 +138,22 @@ with gr.Blocks() as app:
 
     def clear_data():
         return {
-            top_pred_image: None,
-            top_class_output_img: None
+            pred_image: None,
+            output_img: None,
+            grad_cam_out: None
         }
 
 
-    tc_clear_btn.click(clear_data, None, [top_pred_image, top_class_output_img])
+    clear_btn.click(clear_data, None, [pred_image, output_img])
 
 
     def on_select(evt: gr.SelectData):
         return {
-            top_pred_image: sample_images[evt.index][0]
+            pred_image: sample_images[evt.index][0]
         }
 
 
-    example_images.select(on_select, None, top_pred_image)
+    example_images.select(on_select, None, pred_image)
 
 
     def plot_image(image, boxes):
@@ -232,20 +233,20 @@ with gr.Blocks() as app:
         return [plotted_img, cam_image]
 
 
-    def top_class_img_upload(input_img, if_cam):
+    def img_upload(input_img, if_cam):
         if input_img is not None:
             imgs = predict(input_img, show_cam=if_cam)
 
             return {
-                top_class_output_img: imgs[0],
+                output_img: imgs[0],
                 grad_cam_out: imgs[1]
             }
 
 
-    top_class_btn.click(
-        top_class_img_upload,
-        [top_pred_image, if_show_grad_cam],
-        [top_class_output_img, grad_cam_out]
+    submit_btn.click(
+        img_upload,
+        [pred_image, if_show_grad_cam],
+        [output_img, grad_cam_out]
     )
 
 '''