.

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, map_resume):
+def apply_color_map(rgb, color_map=cdl_color_map, map_resume=map_resume):
     
     rgb_mapped = rgb.copy()
     
@@ -271,15 +271,16 @@ with gr.Blocks() as demo:
                 inp3=gr.Image(image_mode='RGB', scale=10, label='T3')
 
     with gr.Row():
-        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')
+        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():
             out = gr.Image(image_mode='RGB', scale=10, label='Model prediction')
             # gr.Image(value='Legend.png', image_mode='RGB', scale=2, show_label=False)