Update channel_functions.py

channel_functions.py

@@ -105,6 +105,7 @@ def which_channel_dominates(img_arr, original_image_plot = 'yes', original_image
     print(np.unique(which_channel_dominates))
 
     #Map the color code to the image
+    # fig, ax = plt.subplots(figsize=(8,8))
     plot = plt.imshow(which_channel_dominates, cmap=cmap, alpha=channel_opacity)
 
     #Customize the ticks of the colorbar
@@ -115,4 +116,5 @@ def which_channel_dominates(img_arr, original_image_plot = 'yes', original_image
                 )
     
     text = "Which channel dominates in the image below?\nWhite : R=G=B=255, Black : R=G=B=0\nGray : 0 < R=G=B< 255"
-    plt.title(text)
+    plt.title(text)
+    plt.plot()