Controling figure sizes in Jupyter's VSC extension

I started using the Jupyter VSC extension and I must say I’m very well impressed. But there is one issue that is a bit annoying, which is the figure sizes. When we have a command that generates a figure, that figure takes the whole width of the page and that’s too big.

Anyone knows if there is a way we can control the size of the figure?