.

app.py

@@ -216,7 +216,7 @@ def inference_on_file(target_image, model, custom_test_pipeline, map_resume):
     
     output=apply_color_map(output,map_resume).transpose((1,2,0))
     bar_data = pd.DataFrame(map_resume)
-    return rgb1,rgb2,rgb3,output
+    return rgb1,rgb2,rgb3,output,bar_data
 
 def process_test_pipeline(custom_test_pipeline, bands=None):
     
@@ -287,7 +287,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')