Adding more layers to your model doesn’t necessarily improve the accuracy so you would need to experiment with your model for your use case.
Based on your code it seems you could simply pass more feature sizes in layers to add more linear layers.