change output type

app.py

@@ -253,7 +253,7 @@ def update_display(image):
     exp_gallery = gr.Gallery(label="Explanation Heatmaps for top 5 predicted classes", show_label=False,elem_id="gallery",columns=[5], rows=[1],height='auto', allow_preview=True, preview=None)
     closest_gallery = gr.Gallery(label="Closest Images", show_label=False,elem_id="gallery",columns=[5], rows=[1],height='auto', allow_preview=True, preview=None)
     diagram= gr.Image(label = 'Bar Chart')
-    return processed_image,processed_image,instruction,model_name,explain_method,sampling_size,top_k,class_predicted,exp_gallery,closest_gallery,diagram
+    return processed_image,processed_image,instruction,model_name,explain_method,sampling_size,top_k,exp_gallery,closest_gallery,diagram
 def update_slider_visibility(explain_method):
     bool = explain_method=="Rise"
     return {sampling_size: gr.Slider(1, 5000, value=2000, label="Sampling Size in Rise", visible=bool, interactive=True)}
@@ -425,7 +425,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
     process_button.click(
                         fn=update_display,
                         inputs=input_image,
-                        outputs=[input_image,workbench_image,instruction_text,model_name,explain_method,sampling_size,top_k,class_predicted,exp_gallery,closest_gallery,diagram]
+                        outputs=[input_image,workbench_image,instruction_text,model_name,explain_method,sampling_size,top_k,exp_gallery,closest_gallery,diagram]
                     )