Если я изменю значение ползунка, график отскакивает

Question

Я строю линейную функцию, и если я изменяю значение ползунка, область графика отскакивает ... я имею в виду исчезновение и появление ... и я хочу исправить эту область графика или окно или что-то еще, потому что мойУчитель сказал мне: попытайтесь это исправить, потому что это немного раздражает.Все хорошо с кодом, только то, что я хочу исправить.Спасибо за вашу помощь вперед!

display(Math('f(x) = ax + b'))

print("Pls add coefficients")

@widgets.interact()
def plot(a=4, b=4 ,grid=True):
    x = np.linspace(-4., +4., 1000)

    if a == 0:
        print("This is not a linear function, because A cant be 0!")
    else:
        x1 = -b/a
        print('X: {}'.format(x1))
    y = (a*x + b)

    fig, ax = plt.subplots(1, 1, figsize=(8, 8))
    ax.plot(x, y)
    plt.xlim(-8, 8)
    plt.ylim(-10,10)
    plt.axhline(0, color="red", lw=1)
    plt.axvline(0, color="red", lw=1)
    plt.xlabel('X tengely')
    plt.ylabel('Y tengely')
    plt.title('Linear function')
    ax.grid(grid)