**# generate predictions on the original dataset**
predictions = predict_model(best, data=data)
**# add a date column in the dataset**
predictions['Date'] = pd.date_range(start='1949-01-01', end = '1960-12-01', freq = 'MS')
fig = px.line(predictions, x='Date', y=["Passengers", "Label"], template = 'plotly_dark')
**# add a vertical rectange for test-set separation**
fig.add_vrect(x0="1960-01-01", x1="1960-12-01", fillcolor="grey", opacity=0.25, line_width=0)fig.show()