.

app.py

@@ -215,8 +215,11 @@ def inference_on_file(target_image, model, custom_test_pipeline, map_resume):
     
     
     output=apply_color_map(output,map_resume).transpose((1,2,0))
-        
-    return rgb1,rgb2,rgb3,output
+    bar_data = pd.DataFrame(map_resume)
+    print("map_resume", map_resume)
+    print("bar_data", bar_data)
+
+    return rgb1,rgb2,rgb3,output,bar_data
 
 def process_test_pipeline(custom_test_pipeline, bands=None):
     
@@ -247,9 +250,6 @@ 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, map_resume=map_resume)
 
-bar_data = pd.DataFrame(map_resume)
-print("map_resume", map_resume)
-print("bar_data", bar_data)
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -286,7 +286,7 @@ with gr.Blocks() as demo:
             out = gr.Image(image_mode='RGB', scale=10, label='Model prediction')
             # gr.Image(value='Legend.png', image_mode='RGB', scale=2, show_label=False)
   
-    btn.click(fn=func, inputs=inp, outputs=[inp1, inp2, inp3, out])
+    btn.click(fn=func, inputs=inp, outputs=[inp1, inp2, inp3, out,bar_data])
     
     with gr.Row():
         gr.Markdown(value='### Model prediction legend')