The output layer in this example looks like any other multiclass classification. I've included a single dense layer before the output layer as well, as shown in the following code:
x = Dense(128, activation='relu')(x)
preds = Dense(20, activation='softmax')(x)