minor style fix

app.py

@@ -14,7 +14,7 @@ def load_model(model_name):
     return model, model_params, tokenizer
     
 
-col1, col2, col3, *_ = st.columns(5)
+col1, col2, col3, *_ = st.columns(4)
 model_name = col1.selectbox("Select a model: ", options=['gpt2', 'gpt2-medium', 'gpt2-large'])
 model, model_params, tokenizer = load_model(model_name)
 # neuron_layer = col2.text_input("Layer: ", value='0')