Update app.py

app.py

@@ -23,7 +23,7 @@ if st.button("Train"):
     model.compile(optimizer="sgd",loss="sparse_categorical_crossentropy",metrics=["accuracy"])
     model.fit(x_train,y_train,epochs=10,batch_size=128,validation_split=0.2)
     fig, axs = plt.subplots(6, 1, figsize=(8, 6))
-    col1,col2,col3,col5,col6 = st.columns(6)
+    col1,col2,col3,col4,col5,col6 = st.columns(6)
     with col1:
         st.write("Filters")
         for i in range(6):