trying two figures

app.py

@@ -123,6 +123,8 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     
     origin = 'lower'
     
+    
+    ##### Make the activation figure ######
     plt.rcParams['xtick.labelsize'] = ticks
     plt.rcParams['ytick.labelsize'] = ticks
     
@@ -148,9 +150,26 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     ax1.set_title('Activation 1', fontsize=titles)
     ax2.set_title('Activation 2', fontsize=titles)
     
+    
+    ##### make the figure for the input image #####
+    plt.rcParams['xtick.labelsize'] = ticks
+    plt.rcParams['ytick.labelsize'] = ticks
+    
+    input_fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(14, 12))
+    
+    
+    im0 = ax.imshow(input_image, cmap=cmap,
+                     origin=origin)    
+    
+    divider = make_axes_locatable(ax)
+    cax = divider.append_axes('right', size='5%', pad=0.05)
+    fig.colorbar(im0, cax=cax, orientation='vertical')
+        
+    ax.set_title('Input', fontsize=titles)
+    
     print("Sending to Hugging Face")
     
-    return output, fig
+    return output, input_fig, fig
 
 
 if __name__ == "__main__":
@@ -172,6 +191,7 @@ if __name__ == "__main__":
                 gr.File(label="Input Data", show_label=True)],
         outputs=[gr.Textbox(lines=1, label="Prediction", show_label=True), 
                 # gr.Image(label="Input Image", show_label=True), 
+                gr.Plot(label="Input Image", show_label=True), 
                 #  gr.Image(label="Activation 1", show_label=True), 
                 #  gr.Image(label="Actication 2", show_label=True)],
                 gr.Plot(label="Activations", show_label=True)