how to set vertical align middle in div code example

Example: center elements vertically in div

.container {        
    display: flex;
    align-items: center;
}