Update app.py

app.py

@@ -66,4 +66,4 @@ else:
 
 with st.expander("See first convolutional layer"):
     rgb = np.repeat(np.expand_dims(np.array(params["Conv_0"]["kernel"].squeeze()[:,:,0]+1)*127., axis=0), repeats=3, axis=0)
-    st.image(Image.fromarray(rgb, mode="RGB").resize((28, 28))
+    st.image(Image.fromarray(rgb, mode="RGB").resize((28, 28)))