Update app.py

app.py

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