Update app.py

app.py

@@ -72,4 +72,5 @@ with st.expander("See first convolutional layer"):
     gridify(params["Conv_0"]["kernel"], grid=(4,8), kernel_size=(3,3))
     
 with st.expander("See second convolutional layer"):
+    print(params["Conv_1"]["kernel"].shape)
     gridify(params["Conv_1"]["kernel"], grid=(8,8), kernel_size=(3,3))