Making plot with subfigure created elsewhere (matplotlib)

Question

I have a function "makeGrid(tensor)" that takes as an argument a pytorch tensor of shape (B,3,H,W), and uses ImageGrid to make a figure that displays the batch of figures in a grid.

Now, the model that outputs "tensor", depends on one parameter, "alpha". I would like to include a slider on the figure such that I can modify alpha "live". I am using the "Slider" widget from matplotlib roughly as such :

result = model(tensor)
images,grid = makeGrid(result)

ifig = plt.figure()# Figure with slider     
axalpha = ifig.add_axes([0.25, 0.1, 0.65, 0.03])
# How to add the "images" to ifig ???

    
alpha_slider = Slider(
    ax=axsmol,
    valmin=-2,
    valmax=2,
    valinit=1,
)


def update(val):
    model.alpha = alpha_slider.val
    result= model(img_batch) 
    images,grid = makeGrid(result)
    # Same problem, need to update ifig with new images

alpha_slider.on_changed(update)
plt.show()

So, my main problem is I have no idea how to use the already created figure (images) and/or grid (which is an ImageGrid object, roughly a list of axes afaik) as a subplot of "ifig", the interactive figure which contains slider and images.

Very sorry as this seems to be a basic question, but searching for "how to add already created figure as subplot of figure" or other things didn't yield solutions to my problem (or at least, in my limited point of view).