Отображение последовательности изображений в Jupyter Notebook

Я использовал следующую функцию для отображения последовательности изображений:

import matplotlib.pyplot as plt
import matplotlib.animation as animation
from IPython.display import HTML

def showSample(input_sample):
    numFrames = input_sample.shape[-1]
    fig = plt.figure()
    im = plt.imshow(input_sample[:,:,0])

    def update(i):
        img = input_sample[:,:,i]
        im.set_data(img)
        return im

    ani = animation.FuncAnimation(fig, update, frames=numFrames, repeat=False)  
    display(HTML(ani.to_html5_video()))

В результате получается анимированная последовательность изображений по желанию, за которой следует еще одна фигура кадра 0, вероятно, из строки: im = plt.imshow(input_sample[:,:,0]). Как отобразить только последовательность изображений без дополнительной нежелательной фигуры?

Спасибо.