using figure for activations now

app.py

@@ -1,6 +1,6 @@
 from matplotlib import cm 
 import matplotlib.pyplot as plt
-# from mpl_toolkits.axes_grid1 import make_axes_locatable
+from mpl_toolkits.axes_grid1 import make_axes_locatable
 
 import numpy as np
 
@@ -167,9 +167,35 @@ def predict_and_analyze(model_name, num_channels, dim, image):
     activation_1 = normalize_array(activation_1)
     activation_2 = normalize_array(activation_2)
     
+    print("Plotting")
+    
+    origin = 'lower'
+    
+    plt.rcParams['xtick.labelsize'] = ticks
+    plt.rcParams['ytick.labelsize'] = ticks
+    
+    fig, axs = plt.subplots(nrows=1, ncols=2, figsize=(19, 8))
+    
+    ax1, ax2 = axs[0], axs[1]
+    
+    im1 = ax1.imshow(activation_1, cmap=cmap, #vmin=0, vmax=1, 
+                     origin=origin)
+    im2 = ax2.imshow(activation_2, cmap=cmap, #vmin=0, vmax=1, 
+                     origin=origin) 
+    
+    ims = [im1, im2]
+    
+    for (i, ax) in enumerate(axs):
+        divider = make_axes_locatable(ax)
+        cax = divider.append_axes('right', size='5%', pad=0.05)
+        fig.colorbar(ims[i], cax=cax, orientation='vertical')
+        
+    ax1.set_title('First Activation', fontsize=titles)
+    ax2.set_title('Second Activation', fontsize=titles)
+    
     print("Sending to Hugging Face")
     
-    return output, input_image, activation_1, activation_2
+    return output, input_image, fig
 
 
 demo = gr.Interface(
@@ -189,8 +215,11 @@ demo = gr.Interface(
             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.Image(label="Activation 1", show_label=True), 
-             gr.Image(label="Actication 2", 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) 
+            #  gr.Plot(label="Actication 2", show_label=True)],
+            ],
     title="Kinematic Planet Detector"
 )
 demo.launch()