Update app.py

app.py

@@ -67,4 +67,4 @@ else:
 with st.expander("See first convolutional layer"):
     grid = np.array(params["Conv_0"]["kernel"].reshape((3, 3, 4, 8)).transpose(2,0,3,1).reshape(12, 24)+1)*127.
     rgb = np.repeat(np.expand_dims(grid, axis=0), repeats=3, axis=0).astype(np.uint8).transpose(1,2,0)
-    st.image(Image.fromarray(rgb, mode="RGB").resize((24*4, 24*8)))
+    st.image(Image.fromarray(rgb, mode="RGB"))