.

app.py

@@ -57,7 +57,7 @@ cdl_color_map = [{'value': 1, 'label': 'Natural vegetation', 'rgb': (233,255,190
                  {'value': 12, 'label': 'Sorghum', 'rgb':(255,158,15), 'qtt': 0},
                  {'value': 13, 'label': 'Other', 'rgb':(0,175,77), 'qtt': 0}]
 
-def apply_color_map(rgb, color_map=cdl_color_map):
+def apply_color_map(rgb, color_map=cdl_color_map, map_resume=map_resume):
     
     rgb_mapped = rgb.copy()
     
@@ -68,8 +68,8 @@ def apply_color_map(rgb, color_map=cdl_color_map):
             cont = np.where((rgb[0] == map_tmp['value']) & (rgb[1] == map_tmp['value']) & (rgb[2] == map_tmp['value']), 1, 0)
     
         print(map_tmp['label'])
-        #print("Cantidad total ", sum(sum(cont)))
-        map_resume.append({"label": map_tmp['label'],'qtt': sum(sum(cont))})
+        print("Cantidad total ", sum(cont))
+        map_resume.append({"label": map_tmp['label'],'qtt': sum(cont)})
         
     return rgb_mapped    
     
@@ -180,7 +180,7 @@ def process_rgb(input, mask, indexes):
 
     return rgb
 
-def inference_on_file(target_image, model, custom_test_pipeline):
+def inference_on_file(target_image, model, custom_test_pipeline, map_resume):
 
     target_image = target_image.name
     time_taken=-1
@@ -214,7 +214,7 @@ def inference_on_file(target_image, model, custom_test_pipeline):
     output = np.vstack([output[None], output[None], output[None]]).astype(np.uint8)
     
     
-    output=apply_color_map(output).transpose((1,2,0))
+    output=apply_color_map(output,map_resume).transpose((1,2,0))
         
     return rgb1,rgb2,rgb3,output
 
@@ -245,7 +245,7 @@ model = init_segmentor(config, ckpt, device='cpu')
 custom_test_pipeline=process_test_pipeline(model.cfg.data.test.pipeline, None)
 
 
-func = partial(inference_on_file, model=model, custom_test_pipeline=custom_test_pipeline)
+func = partial(inference_on_file, model=model, custom_test_pipeline=custom_test_pipeline, map_resume=map_resume)
 print("")
 bar_data = pd.DataFrame(map_resume)
 
@@ -271,16 +271,15 @@ with gr.Blocks() as demo:
                 inp3=gr.Image(image_mode='RGB', scale=10, label='T3')
 
     with gr.Row():
-        if len(map_resume) > 1:
-            with gr.Column():
-                gr.BarPlot(bar_data,
-                        x="label",
-                        y="qtt",
-                        title="Simple Bar Plot with made up data",
-                        tooltip=["label", "qtt"])
-                gr.LinePlot(bar_data,
-                            x='label',
-                            y='qtt')
+        with gr.Column():
+            gr.BarPlot(bar_data,
+                    x="label",
+                    y="qtt",
+                    title="Simple Bar Plot with made up data",
+                    tooltip=["label", "qtt"])
+            gr.LinePlot(bar_data,
+                        x='label',
+                        y='qtt')
         with gr.Column():
             out = gr.Image(image_mode='RGB', scale=10, label='Model prediction')
             # gr.Image(value='Legend.png', image_mode='RGB', scale=2, show_label=False)